fixed db issues

1 files changed, 39 insertions, 0 deletions

diff --git a/examples/db_interact.py b/examples/db_interact.py
new file mode 100644
index 0000000..7ab6810
--- /dev/null
+++ b/examples/db_interact.py
@@ -0,0 +1,39 @@
+#!/usr/bin/python2.7
+import sys
+sys.path.append('./lib')
+from db import Db, ImCmd
+import simplejson as json
+import re
+db = Db()
+def set_tag_generate(imcmd, tag="PbGenerate"):
+    dataobj = json.loads(imcmd.dataobj)
+    if dataobj.get("background"):
+        tag = dataobj.get("compose")
+    elif dataobj.get("transparent") != 'false':
+        tag = dataobj.get("subtract")
+    imcmd.tag = tag
+    session.add(imcmd)
+
+def set_tag_pattern(imcmd, tag="PbPattern"):
+    imcmd.tag = tag
+    session.add(imcmd)
+session = db.Session()
+
+
+imcmd_list = session.query(ImCmd).filter_by(tag="PbGenerate").all()
+for i in imcmd_list:
+    set_tag_generate(i)
+session.commit()
+
+imcmd_list = session.query(ImCmd).filter_by(tag="dither").all()
+for i in imcmd_list:
+    set_tag_pattern(i)
+session.commit()
+
+imcmd_list = session.query(ImCmd).filter_by(tag="im").all()
+for i in imcmd_list:
+    set_tag_generate(i)
+session.commit()
+
+#taglist = list(set(map(lambda n: n.tag, imcmd_list)))
+#print taglist