Updated login to work correctly with current spline. #69

[zzz-spline-users.git] / migration / versions / 001_Add_user_and_openid_tables.py

diff --git a/migration/versions/001_Add_user_and_openid_tables.py b/migration/versions/001_Add_user_and_openid_tables.py
new file mode 100644 (file)
index 0000000..473d3df
--- /dev/null
+++ b/migration/versions/001_Add_user_and_openid_tables.py
@@ -0,0 +1,25 @@
+from sqlalchemy import *
+from migrate import *
+
+from sqlalchemy.ext.declarative import declarative_base
+TableBase = declarative_base(bind=migrate_engine)
+
+
+class User(TableBase):
+    __tablename__ = 'users'
+    id = Column(Integer, primary_key=True)
+    name = Column(Unicode(length=20), nullable=False)
+
+class OpenID(TableBase):
+    __tablename__ = 'openid'
+    openid = Column(Unicode(length=255), primary_key=True)
+    user_id = Column(Integer, ForeignKey('users.id'))
+
+
+def upgrade():
+    User.__table__.create()
+    OpenID.__table__.create()
+
+def downgrade():
+    OpenID.__table__.drop()
+    User.__table__.drop()