1 files changed, 40 insertions, 0 deletions

diff --git a/db.py b/db.py
new file mode 100644
index 0000000..5e13924
--- /dev/null
+++ b/db.py
@@ -0,0 +1,40 @@
+import sqlalchemy
+from sqlalchemy import *
+from sqlalchemy.orm import sessionmaker
+
+def setupdb():
+        #engine = create_engine('sqlite:///Users/erdgeist/Coding/tinybill/foo.db', echo=True)
+        engine = create_engine('sqlite:///foo.db', echo=True)
+        metadata = MetaData()
+        
+        Table( 'customers', metadata,
+                Column( 'id', Integer, primary_key=True ),
+                Column( 'name', String, unique=True ),
+                Column( 'email', String )
+        )
+        
+        Table( 'products', metadata,
+                Column( 'id', Integer, primary_key=True ),
+                Column( 'type', String ),
+                Column( 'name', String, unique=True ),
+                Column( 'price', Integer ),
+        )
+        
+        Table( 'purchases', metadata,
+                Column( 'id', Integer, primary_key=True ),
+                Column( 'product', Integer, ForeignKey('products.id' ) ),
+                Column( 'customer', Integer, ForeignKey('customers.id' ) ),
+                Column( 'detail', String ),
+                Column( 'date', Date )
+        )
+        
+        Table( 'payment', metadata,
+                Column( 'id', Integer, primary_key=True ),
+                Column( 'purchase', Integer, ForeignKey('purchases.id') ),
+                Column( 'amount', Integer ),
+                Column( 'date', Date )
+        )
+        
+        metadata.create_all(engine)
+        global Session
+        Session = sessionmaker(bind=engine)