# -*- coding: utf-8 -*-

from database import Database

class Acounting(Database):
    def __init__(self,id, connection):
        Database.__init__(self,connection)
        self.id=id

    def setId(self,id):
        self.id=id

    def getStatus(self):
        self.connect()
        self.cursor.execute ("SELECT status,tel FROM %s WHERE id='%s'" % (self.connection['table'], self.id))
        ret= self.cursor.fetchall()
        self.disconnect()
        return ret

    def addGood(self, good,disconnect=True):
        if type(good) == list:
            for i in good:
                self.addGood(i)
            if disconnect:
                self.disconnect()
        else:
            self.connect()
            self.cursor.execute("INSERT INTO %s (id,tel,status) VALUES('%s','%s','sended')" % (self.connection['table'], self.id, good))
            if disconnect:
                self.disconnect()
            

    def addFailed(self, failed,disconnect=True):
        if type(failed) == list:
            for i in failed:
                self.addFailed(i,False)
            if disconnect:
                self.disconnect()
        else:
            self.connect()
            self.cursor.execute ("INSERT INTO %s (id,tel,status) VALUES('%s','%s','failed')"%(self.connection['table'], self.id, failed))
            if disconnect:
                self.disconnect()