1 files changed, 13 insertions, 5 deletions
diff --git a/examples/db_interact.py b/examples/db_interact.py
index 7ab6810..d60037f 100644
--- a/examples/db_interact.py
+++ b/examples/db_interact.py
@@ -3,20 +3,23 @@ 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")
+        tag = "%s:%s" % ("PbGenerate", dataobj.get("compose"))
     elif dataobj.get("transparent") != 'false':
-        tag = dataobj.get("subtract")
+        tag = "PbGenerate:transparent"
     imcmd.tag = tag
     session.add(imcmd)
 
+
 def set_tag_pattern(imcmd, tag="PbPattern"):
     imcmd.tag = tag
     session.add(imcmd)
+
 session = db.Session()
 
 
@@ -35,5 +38,10 @@ for i in imcmd_list:
     set_tag_generate(i)
 session.commit()
 
-#taglist = list(set(map(lambda n: n.tag, imcmd_list)))
-#print taglist
+imcmd_list = session.query(ImCmd).filter_by(tag="True").all()
+for i in imcmd_list:
+    set_tag_generate(i)
+session.commit()
+
+# taglist = list(set(map(lambda n: n.tag, imcmd_list)))
+# print taglist