Add Story model.

generate_database.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python
 
 import bcrypt
-from server import db
-from server import User
+import os
+from server import db,app
+from server import User, Story
+
+os.remove("db.sqlite")
 
 db.create_all()
 
@@ -11,4 +14,17 @@
 
 db.session.add(t)
 db.session.add(j)
+
+db.session.commit()
+
+s1 = Story(t.id, 'This is story 1')
+s2 = Story(t.id, 'Story 2 is much cooler!')
+s3 = Story(j.id, 'This, story 3, is superior.')
+s4 = Story(j.id, 'Story 4 is the best! I think.')
+
+db.session.add(s1)
+db.session.add(s2)
+db.session.add(s3)
+db.session.add(s4)
+
 db.session.commit()

server.py

@@ -8,8 +8,9 @@
 
 class User(db.Model):
     id = db.Column(db.Integer, primary_key=True)
-    username = db.Column(db.String(30), unique=True)
+    username = db.Column(db.String(30))
     password = db.Column(db.String(40))
+    posts = db.relationship('Story', backref='user', lazy='dynamic')
 
     def __init__(self, username, password):
         self.username = username
@@ -18,6 +19,18 @@ def __init__(self, username, password):
     def __repr__(self):
         return "<User {0}>".format(self.username)
 
+class Story(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    poster_id = db.Column(db.Integer, db.ForeignKey('user.id'))
+    href = db.Column(db.String(255))
+
+    def __init__(self, poster_id, href):
+        self.poster_id = poster_id
+        self.href = href
+
+    def __repr__(self):
+        return "<Story #{0}>".format(self.href)
+
 @app.route('/')
 @app.route('/top')
 def top_stories():