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

     2 

     3 from database import Database

     4 import logging

     5 logger=logging.getLogger("iro.acounting");

     6 

     7 class Acounting(Database):

     8     def __init__(self,id, connection):

     9         Database.__init__(self,connection)

    10         self.id=id

    11 

    12     def setId(self,id, user):

    13         self.id=id

    14         if user:

    15             self.connect()

    16             self.cursor.execute ("INSERT INTO %s (id,user) VALUES ('%s','%s')" % (self.connection['overview'], self.id, user))

    17             self.disconnect()

    18 

    19 

    20     def getStatus(self):

    21         self.connect()

    22         self.cursor.execute ("SELECT status,tel FROM %s WHERE id='%s'" % (self.connection['table'], self.id))

    23         ret= self.cursor.fetchall()

    24         self.disconnect()

    25         return ret

    26 

    27     def addGood(self, good,disconnect=True):

    28         if type(good) == list:

    29             for i in good:

    30                 self.addGood(i)

    31             if disconnect:

    32                 self.disconnect()

    33         else:

    34             self.connect()

    35             self.cursor.execute("INSERT INTO %s (id,tel,status) VALUES('%s','%s','sended')" % (self.connection['table'], self.id, good))

    36             if disconnect:

    37                 self.disconnect()

    38             

    39 

    40     def addFailed(self, failed,disconnect=True):

    41         if type(failed) == list:

    42             for i in failed:

    43                 self.addFailed(i,False)

    44             if disconnect:

    45                 self.disconnect()

    46         else:

    47             self.connect()

    48             self.cursor.execute ("INSERT INTO %s (id,tel,status) VALUES('%s','%s','failed')"%(self.connection['table'], self.id, failed))

    49             if disconnect:

    50                 self.disconnect()