X-Git-Url: http://git.veekun.com/zzz-spline-users.git/blobdiff_plain/318d3515bfcbe35451db3624b949ccd7a1d66720..1e2382335ea1e52b42cdc07fd1f74b9a9b0cec39:/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
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()