Added some openid code.

flask_website/database.py

@@ -0,0 +1,87 @@
+from datetime import datetime
+from sqlalchemy import create_engine, MetaData, Table, Column, Integer, \
+     String, DateTime, ForeignKey
+from sqlalchemy.orm import scoped_session, sessionmaker, backref
+from sqlalchemy.ext.declarative import declarative_base
+
+from flask_website import config
+
+engine = create_engine(config.DATABASE_URI)
+db_session = scoped_session(sessionmaker(autocommit=False,
+                                         autoflush=False,
+                                         bind=engine))
+
+def init_db():
+    Model.metadata.create_all(bind=engine)
+
+
+class Model(declarative_base()):
+    query = db_session.query_property()
+
+
+class User(Model):
+    __tablename__ = 'users'
+    id = Column('user_id', Integer, primary_key=True)
+    openid = Column('openid', String(200))
+    username = Column(String(40), unique=True)
+    password = Column(String(80))
+
+    def __init__(self, username, password):
+        self.username = username
+        self.password = password
+
+
+class Category(Model):
+    __tablename__ = 'categories'
+    id = Column('category_id', Integer, primary_key=True)
+    name = Column(String(50))
+    slug = Column(String(50))
+
+
+class Snippet(Model):
+    __tablename__ = 'snippets'
+    id = Column('snippet_id', Integer, primary_key=True)
+    author = ForeignKey(User, backref=backref('snippets', lazy='dynamic'))
+    category = ForeignKey(Category, backref=backref('snippets', lazy='dynamic'))
+    title = Column(String(200))
+    body = Column(String)
+    pub_date = DateTime()
+
+    def __init__(self, author, title, body):
+        self.author = author
+        self.title = title
+        self.body = body
+        self.pub_date = datetime.utcnow()
+
+
+class Comment(Model):
+    __tablename__ = 'comments'
+    id = Column('comment_id', Integer, primary_key=True)
+    snippet = ForeignKey(Snippet, backref='lazy')
+    author = ForeignKey(User, backref=backref('comments', lazy='dynamic'))
+    title = Column(String(200))
+    text = Column(String)
+    pub_date = DateTime()
+
+    def __init__(self, author, title, text):
+        self.author = author
+        self.title = title
+        self.text = text
+        self.pub_date = datetime.utcnow()
+
+
+class OpenIDAssociation(Model):
+    id = Column('association_id', Integer, primary_key=True)
+    server_url = Column(String(1024))
+    handle = Column(String(255))
+    secret = Column(String(255))
+    issued = Column(Integer)
+    lifetime = Column(Integer)
+    assoc_type = Column(String(64))
+
+
+class OpenIDUserNonces(Model):
+    id = Column('user_nonce_id', Integer, primary_key=True)
+    server_url = Column(String(1024))
+    timestamp = Column(Integer)
+    salt = Column(String(40))