Esempio n. 1
0
    def __init__(self):
        # self.invoicer = Invoicer()
        # self.pubsub = RedisPubSub("mama")
        self.threadQ = Queue.Queue()
        self.webIntf = WebIntf(self)
        self.solver = DilatorSolver(self.threadQ)
        self.results = []
        self.toNodeQ = []
        self.prevNodeQ = [None] * 1000
        self.dbChanged = True
        pub.subscribe(self.setCurrentPage, "currentPage")
        self.range = 0
        print "Server Listening..."
        while 1:
            if self.dbChanged:
                self.doLpInterval()
                self.dbChanged = False

            time.sleep(30)
            self.doCrtvInterval()
Esempio n. 2
0
class DilatorDaemon(object):
    """
    Main server daemon responsible the LP and sending crtvs
    """

    def __init__(self):
        # self.invoicer = Invoicer()
        # self.pubsub = RedisPubSub("mama")
        self.threadQ = Queue.Queue()
        self.webIntf = WebIntf(self)
        self.solver = DilatorSolver(self.threadQ)
        self.results = []
        self.toNodeQ = []
        self.prevNodeQ = [None] * 1000
        self.dbChanged = True
        pub.subscribe(self.setCurrentPage, "currentPage")
        self.range = 0
        print "Server Listening..."
        while 1:
            if self.dbChanged:
                self.doLpInterval()
                self.dbChanged = False

            time.sleep(30)
            self.doCrtvInterval()

    def setCurrentPage(self, range):
        self.range = int(range)
        # print "Range: "+str(range)+" to "+str(int(range)+15)
        self.doCrtvInterval()

    def doCrtvInterval(self):
        """
        Iterates through toNodeQ, formats, and passes to webintf
        """
        # print "Sent Crtvs"
        self.createSignCrtvQueues()
        self.toNodeQ = sorted(self.toNodeQ, key=lambda k: k.get("sign").id)
        tonode = "["
        for q in self.toNodeQ:
            signid = int(q.get("sign").id)
            if signid >= self.range and signid <= self.range + 15:
                tonode = tonode + self.formatJsonMsg(q.get("sign"), q.get("currCrtv"), q.get("cmpgn")) + ","
        tonode = tonode[:-1] + "]"
        self.webIntf.publish(tonode)

    def doLpInterval(self):
        """
        DocString
        """
        print "Lp Interval"
        self.toNodeQ = []
        self.lpLock = 1
        self.solver.run()
        self.formatProb(self.threadQ.get())
        self.lpLock = 0

    def formatProb(self, prob):
        for v in prob:
            if v[1] == "NULL":
                cmpgn = "NULL"
            else:
                cmpgn = self.solver.cmpgns.get(id=v[1])
            sign = self.solver.signs.get(id=v[0])
            self.results.append({"cmpgn": cmpgn, "sign": sign})

    def createSignCrtvQueues(self):
        """
        For each Sign/Cmpgn match, assaign a random creative to the sign
        and append it to the queue.
        """
        for i, r in enumerate(self.results):
            signdisp = r.get("sign").signdisp

            if not r.get("cmpgn") == "NULL":
                crtvs = self.filterValidCrtvs(r, signdisp)
                currCrtv = crtvs[random.randint(0, len(crtvs) - 1)]
            else:
                currCrtv = "NULL"

            if self.prevNodeQ[i] != {"sign": r.get("sign"), "currCrtv": currCrtv, "cmpgn": r.get("cmpgn")}:
                self.toNodeQ.append({"sign": r.get("sign"), "currCrtv": currCrtv, "cmpgn": r.get("cmpgn")})
            self.prevNodeQ[i] = {"sign": r.get("sign"), "currCrtv": currCrtv, "cmpgn": r.get("cmpgn")}
        self.results = []

    def filterValidCrtvs(self, r, signdisp):
        """
        DocSring
        """
        if signdisp == 1:
            return [crtv for crtv in r.get("cmpgn").crtv.all() if crtv.type == "poster"]
        if signdisp == 2:
            return [crtv for crtv in r.get("cmpgn").crtv.all() if crtv.type == "video"]
        if signdisp == 3:
            return [crtv for crtv in r.get("cmpgn").crtv.all()]

    def formatJsonMsg(self, sign, crtv, cmpgn):
        """
        Json to be sent to the webinterface
        """
        cropname = "c" + str(sign.ypixel)
        if not crtv == "NULL":
            return json.dumps(
                {
                    "cmpgnname": cmpgn.name,
                    "signid": sign.id,
                    "username": cmpgn.user.username,
                    "crtvfile": os.path.basename(crtv.file.path),
                    "crtvtype": crtv.type,
                    "cropname": cropname,
                }
            )
        else:
            return json.dumps({"signid": sign.id, "crtvfile": "NULL"})

    @staticmethod
    def formatInvoice(sign, crtv, cmpgn):
        """
        Formatting to be sent to the invoicer
        """
        return {"cmpgnname": cmpgn.name, "signid": sign.id, "crtvid": crtv.id}

    def printOut(self):
        """
        Generic print to stout for debugging
        """
        name = (v.name for v in self.prob.variables() if v.varValue)
        for n in name:
            print n

    def printCmpgnTally(self, r):
        """
        DocSring
        """
        prevcmpgn, count, total = 1, 0, 0
        if prevcmpgn == r.get("cmpgn").id:
            count = count + 1
            total = total + 1
        else:
            print prevcmpgn, count, total
            count = 0
        prevcmpgn = r.get("cmpgn").id

        print prevcmpgn, count, total