Beispiel #1
0
 def __init__(self, *args, **kwargs):
     try:
         self.db = Connection(*args, **kwargs)
     except:
         self.db = None
Beispiel #2
0
class AccountModel(object):
    def __init__(self, *args, **kwargs):
        try:
            self.db = Connection(*args, **kwargs)
        except:
            self.db = None

    def __setitem__(self, account, state):
        acc = self.db.accounts
        try:
            acc.insert(account=account, state=state)
        except IntegrityError:
            if state == -1:
                sql = "update accounts set state = %(state)d where account='%(account)s'" % locals()
            else:
                sql = "update accounts set state = %(state)d where account='%(account)s' and state < %(state)d and state != -1" % locals()
            self.db.execute(sql)
        return True
    
    def get_action_num(self, action=None):
        if action is None:
            return 0
        row = self.db.get("select * from actions where action='%s' limit 1" % action)
        if row is None:
            return 0
        return row['num']
    
    def update_action_num(self, action, num):
        if action is None:
            return
        try:
            sql = "insert into actions (action, num) values ('%s', %s)" % (action, num)
            self.db.execute(sql)
        except IntegrityError:
            sql = "update actions set num = %d where action='%s'" % (num, action)
            self.db.execute(sql)

    def get_accounts(self, begin=None, state=None, create_time=None, limit = 10000, **kwargs):
        log.trace('get_accounts %s ', begin)
        params = []
        sql = "select * from accounts where "
        if begin is not None:
            sql += 'id > %d' % begin
        
        if state is not None:
            if state < 1:
                sql += ' and state = %d' % state
            else:
                sql += ' and state >= %d' % state
        else:
            sql += ' and state != -1'

        if create_time is not None:
            sql += ' and create_time <= %s '
            params.append(create_time)
        
        for k, v in kwargs.items():
            if v is None:
                continue
            sql += ' and %s = %%s ' % (k)
            params.append(v)

        sql += ' order by id limit %d' % limit
        log.debug("sql %s params %s", sql, params)
        return self.db.query(sql, *params)

    def get_iter(self, action=None, state=None, *args, **kwargs):
        last_id = self.get_action_num(action)
        log.trace('get_iter for action %s from %s', action, last_id)
        while True:
            had_item = False
            
            for row in self.get_accounts(last_id, state, *args, **kwargs):
                yield row
                last_id = row.id
                had_item = True
            if not had_item:
                self.update_action_num(action, 0);
                break 

    def set_can_add(self, limit=0):
        if limit > 0:
            sql = "update accounts set can_add=1 where can_add=0 and state>=1 and state<10 order by id limit %d" % limit
            self.db.execute(sql)
            sql = "update accounts set can_add=0 where can_add=1 and state>=50"
            self.db.execute(sql)
        sql = "select count(id) as num from accounts where can_add=1 and  state>=1"
        can_add = int(self.db.get(sql)['num'])
        sql = "select count(id) as num from accounts where can_add=0 and state=1"
        new_active = int(self.db.get(sql)['num'])
        sql = "select count(id) as num from accounts where state=0"
        new_acc = int(self.db.get(sql)['num'])
        sql = "select count(id) as num from accounts where state>=10"
        can_send = int(self.db.get(sql)['num'])
        sql = "select count(id) as num from accounts where state=-1"
        login_fail = int(self.db.get(sql)['num'])
        return {'can_add' : can_add, 'new_active' : new_active, 'new_acc' : new_acc, 'can_send' : can_send, 'login_fail' : login_fail}