forms

3 files changed, 22 insertions, 1 deletions

diff --git a/cli/app/sql/models/graph.py b/cli/app/sql/models/graph.py
index 59d55c8..ef5a817 100644
--- a/cli/app/sql/models/graph.py
+++ b/cli/app/sql/models/graph.py
@@ -5,7 +5,6 @@ from wtforms_alchemy import ModelForm
 
 from app.sql.common import db, Base, Session
 
-from app.utils.file_utils import sha256_tree
 from app.settings import app_cfg
 
 from os.path import join
@@ -33,3 +32,10 @@ class Graph(Base):
       'created_at': self.created_at,
       'updated_at': self.updated_at,
     }
+
+class GraphForm(ModelForm):
+  class Meta:
+    model = Graph
+    exclude = ['settings', 'created_at', 'updated_at']
+  def get_session():
+    return Session()
diff --git a/cli/app/sql/models/page.py b/cli/app/sql/models/page.py
index 09470b5..1362bb3 100644
--- a/cli/app/sql/models/page.py
+++ b/cli/app/sql/models/page.py
@@ -32,3 +32,10 @@ class Page(Base):
       'created_at': self.created_at,
       'updated_at': self.updated_at,
     }
+
+class PageForm(ModelForm):
+  class Meta:
+    model = Page
+    exclude = ['graph_id', 'settings', 'created_at', 'updated_at']
+  def get_session():
+    return Session()
diff --git a/cli/app/sql/models/tile.py b/cli/app/sql/models/tile.py
index 7ca311d..01b3fab 100644
--- a/cli/app/sql/models/tile.py
+++ b/cli/app/sql/models/tile.py
@@ -33,3 +33,11 @@ class Tile(Base):
       'created_at': self.created_at,
       'updated_at': self.updated_at,
     }
+
+
+class TileForm(ModelForm):
+  class Meta:
+    model = Tile
+    exclude = ['graph_id', 'page_id', 'target_page_id', 'settings', 'created_at', 'updated_at']
+  def get_session():
+    return Session()