from flask import Flask

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:root@localhost/test'

db = SQLAlchemy(app)

class User(db.Model):

id = db.Column(db.Integer, primary_key=True)

username = db.Column(db.String(32), unique=True)

password = db.Column(db.String(32))

def __init__(self, username, password):

self.username = username

self.password = password

def add(self):

try:

db.session.add(self)

db.session.commit()

return self.id

except Exception, e:

db.session.rollback()

return e

finally:

return 0

def isExisted(self):

temUser = User.query.filter_by(username=self.username, password=self.password).first()

if temUser is None:

return 0

else:

return 1

class Entry(db.Model):

id = db.Column(db.Integer, primary_key=True)

content = db.Column(db.Text)

sender = db.Column(db.String(32))

def __init__(self, content, sender):

self.content = content

self.sender = sender

def add(self):

try:

db.session.add(self)

db.session.commit()

return self.id

except Exception, e:

db.session.rollback()

return e

finally:

return 0

def getAllEntry():

Entlist = []

Entlist = Entry.query.filter_by().all()

return Entlist

def test():

db.create_all()

e = Entry("hi", "ErrareHest")

e.add()

l = getAllEntry()

for each in l:

print each

if __name__ == '__main__':

test()