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