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