fixed db issues

2 files changed, 41 insertions, 1 deletions

diff --git a/config.py b/config.py
index 4eed880..a31d091 100644
--- a/config.py
+++ b/config.py
@@ -13,7 +13,7 @@ DEFAULT_FINALFORMAT = "png";
 DEFAULT_HEIGHT = 400
 DEFAULT_WIDTH = 600
 
-OUTPUT_IMAGE_TYPES = ["png", "jpg", "gif" ] 
+OUTPUT_IMAGE_TYPES = ["png", "jpg", "gif" ]
 
 #mounted on tmpfs
 WORKING_DIR = "/var/www/cache"
@@ -32,6 +32,7 @@ SPECIAL_DOWNLOADERS_MAX_SIZE = 100000
 
 
 #database
+#DB_HOST = "localhost"
 DB_HOST = "lalalizard.com"
 DB_USER = "asdfus"
 DB_PASSWORD = "gTYgT&M6q"
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