#!/usr/bin/python2.7
import sys
sys.path.append('./lib')
from db import Db, ImCmd
import simplejson as json
db = Db()


def set_tag_generate(imcmd, tag="PbGenerate"):
    dataobj = json.loads(imcmd.dataobj)
    if dataobj.get("background"):
        tag = "%s:%s" % ("PbGenerate", dataobj.get("compose"))
    elif dataobj.get("transparent") != 'false':
        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()


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()

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