def init(self, options):
self.db = AccountModel(**mysql_config)
try:
self.tos_data = TosData()
except:
pass
for k in dir(self):
v = getattr(self, k, None)
if callable(v) and k.endswith('_iter'):
self.__dict__['_%s' % k] = v()
class AccountServer(MagicServer):
def init(self, options):
self.db = AccountModel(**mysql_config)
try:
self.tos_data = TosData()
except:
pass
for k in dir(self):
v = getattr(self, k, None)
if callable(v) and k.endswith('_iter'):
self.__dict__['_%s' % k] = v()
def add_account(self, handler, names=[], state=0, kwargs={}):
if not isinstance(names, list):
names = [names]
for name in names:
log.trace('add account %s state %s', name, state)
self.db[name] = state
for k, v in kwargs.iteritems():
log.trace('add account %s state %s', k, v)
self.db[k] = v
return True
def new_account_iter(self):
while True:
for row in self.db.get_iter(state=0):
yield row.account
yield None
def get_new_account(self, handler):
try:
ret = self._new_account_iter.next()
log.trace('get new account %s', ret)
return ret
except:
return None
def add_friends_iter(self):
while True:
for row in self.db.get_iter('add_friends', state=1, can_add=1):
yield row
yield None
def get_add_friends(self, handler, limit=10000):
res = []
try:
while True:
row = self._add_friends_iter.next()
if row is None:
break
res.append(row.account)
if len(res) >= limit:
break
self.db.update_action_num('add_friends', row.id)
return res
except StopIteration:
self.db.update_action_num('add_friends', 0)
pass
except:
log.exception('')
return res
def send_accounts_iter(self):
while True:
for row in self.db.get_iter('send_accounts', state=10):
yield row
yield None
def get_send_accounts(self, handler, limit=10000):
res = []
try:
while True:
row = self._send_accounts_iter.next()
if row is None:
break
res.append(row.account)
if len(res) >= limit:
break
self.db.update_action_num('send_accounts', row.id)
except StopIteration:
self.db.update_action_num('send_accounts', 0)
pass
except:
log.exception('')
pass
return res
def set_can_add(self, handler, limit=0):
return self.db.set_can_add(limit)
def get_tos_data(self, handler, limit=10000):
res = []
for i in xrange(limit):
try:
res.append(self.tos_data.get())
except:
log.exception('')
break
self.tos_data.sync()
return res