def get_clients_by_region(self, targeting_code, nums):
try:
if targeting_code[:2] == "CN": # 省
clients = ClientMasterModel.select(ClientMasterModel.id)
return targeting_code, clients.where(ClientMasterModel.province_code == targeting_code).\
order_by(fn.Rand()).limit(nums), nums
else:
province_code = "CN_{code}".format(code=targeting_code[:2])
clients = ClientMasterModel.select(ClientMasterModel.id)
return targeting_code, clients.where(ClientMasterModel.province_code == province_code, ClientMasterModel.city_code == targeting_code).\
order_by(fn.Rand()).limit(nums), nums
except Exception as e:
self.logger.warning(e)
return None
def count(self):
clients = ClientMasterModel.select()
return clients.count()
