From 70a5a9990923bba3c3d6b7b56b561c2b3dcd74ae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Barna?= <robaganrab@gmail.com>
Date: Fri, 28 Oct 2016 05:26:16 +0200
Subject: [PATCH] Add simple Duck model

---
 app.py    |  6 ++++++
 models.py | 15 +++++++++++++++
 2 files changed, 21 insertions(+)
 create mode 100644 models.py

diff --git a/app.py b/app.py
index 4393442..e1900e8 100644
--- a/app.py
+++ b/app.py
@@ -7,8 +7,14 @@ import sqlalchemy
 from sqlalchemy import create_engine, Column, Integer, String
 from sqlalchemy.ext.declarative import declarative_base
 
+Base = declarative_base()
+
+import models
+
 engine = create_engine( os.environ.get( 'DATABASE_URL', 'sqlite:///:memory:' ), echo=True )
 
+Base.metadata.create_all( engine )
+
 app = connexion.App(__name__, specification_dir='./swagger/')
 app.add_api('swagger.yaml',
             arguments={
diff --git a/models.py b/models.py
new file mode 100644
index 0000000..1d2229a
--- /dev/null
+++ b/models.py
@@ -0,0 +1,15 @@
+from sqlalchemy import Column, Integer, String
+from sqlalchemy.ext.declarative import declarative_base
+
+Base = declarative_base()
+
+class Duck( Base ):
+    __tablename__ = 'users'
+
+    duck_id = Column( Integer, primary_key=True )
+    name = Column( String )
+    color = Column( String )
+
+    def __repr__(self):
+        return "<User(name='%s', fullname='%s', password='%s')>" % (
+                             self.name, self.fullname, self.password)