def expire_at_today(self, key, hour=0):
from tools_lib.common_util.archived.gtz import TimeZone
tomorrow = TimeZone.increment_days(TimeZone.local_now())
self.expireat(
key,
TimeZone.increment_hours(
TimeZone.transfer_datetime_to_beginning(tomorrow), hour))
def expire_at_this_end_of_month(self, key):
# 本月月底过期
from tools_lib.common_util.archived.gtz import TimeZone
local_now = TimeZone.local_now()
end_of_this_month = TimeZone.utc_to_local(
TimeZone.month_range(local_now.year, local_now.month)[1])
self.expireat(key, end_of_this_month)
self.delete()
def sign_in():
sign_in = client['profile']['sign_in']
start, end = TimeZone.day_range(value=TimeZone.local_now())
pprint(
"姓名",
"电话",
"时间",
"地址",
"POI",
"经度",
"维度",
)
for doc in sign_in.find({"create_time": {"$gte": start, "$lte": end}}):
pprint(doc['name'], doc['tel'],
TimeZone.utc_to_local(doc['create_time']), doc['loc']['addr'],
doc['loc']['name'], doc['loc']['lng'], doc['loc']['lat'])
def api_deliveryman_sign_in():
"""
派件系人员签到
:param man_id: 根据 id 查出 name, tel, avatar
:param loc: {name, addr, lng, lat}
:param device: {mac_id, ...}
:return: 失败返回400和message
"""
params = ctx.request.input()
# ==> 根据id拿基本信息
man = Man.objects(id=params.man_id).first()
# ==> 检查今日签到次数
now = TimeZone.local_now()
start_time, _end_time = TimeZone.day_range(value=now)
start_time = TimeZone.datetime_to_str(start_time)
sign_in_count = SignIn.objects(**{
'man_id': params.man_id,
'create_time__gte': start_time
}).count()
if sign_in_count >= 50:
logging.warn("Man[%s][%s] try to sign in over 10 times." %
(params.man_id, man.name))
raise ValueError("一天最多签到10次")
# ==> 记录这次签到
name = man.name if man.name else ''
avatar = man.avatar if man.avatar else ''
SignIn(man_id=params.man_id,
name=name,
tel=man.tel,
avatar=avatar,
loc=params.loc,
device=params.device,
create_time=now).save(force_insert=True)
return {
'create_time': TimeZone.datetime_to_str(now,
pattern='%Y-%m-%dT%H:%M:%S')
}
def api_retrieve_man_info(what):
what = str(what).upper().strip()
request = ctx.request.input()
# 如果传入what,by不是允许的值,直接报错
if what not in WHATS:
raise ValueError("WHAT[%s] should be in WHATS%s." %
(what, list(WHATS.keys())))
params = set([str(k).lower().strip() for k in request])
bys = set(BYS.keys())
by = params.intersection(bys)
if not by:
raise ValueError("BY%s should be in BYS%s." %
(list(params), list(bys)))
by_key = next(iter(by))
by_val = request[by_key]
if by_key == 'id' and by_val and len(by_val) != 24:
raise ValueError('find man with id=%s, id should be of length 24.' %
by_val)
# 判断是要什么类型的返回
if what == WHATS.BASIC:
excludes = ('accounts', 'create_time', 'my_man')
man = Man.objects(**{by_key: by_val}).exclude(*excludes).first()
if not man:
raise ValueError("Can't find man with[%s]=[%s]." %
(by_key, by_val))
packed = ManLogic.pack_man(man, excludes=excludes)
ret_cnt = ('familiars', 'pick_up_locations')
for k in ret_cnt:
if k in packed:
packed[k] = len(packed[k]) if packed[k] else 0
return packed
elif what == WHATS.ACCOUNT_LIST:
man = Man.objects(**{by_key: by_val}).only('accounts').first()
return man.accounts
elif what == WHATS.FAMILIAR_LIST:
man = Man.objects(**{by_key: by_val}).only('familiars').first()
return man.familiars
elif what == WHATS.STATUS:
man = Man.objects(**{by_key: by_val}).only('status').first()
return man.status
elif what == WHATS.MAN_LIST:
man = Man.objects(**{by_key: by_val}).only('my_man').first()
if man and man.my_man:
only = ('status', 'id', 'name', 'tel', 'avatar')
my_man = {m['id']: m['bind_time'] for m in man.my_man}
mans = Man.objects(id__in=list(my_man.keys())).only(*only)
ret = []
for m in mans:
packed_man = ManLogic.pack_man(m, only=only)
packed_man['bind_time'] = my_man[str(m.pk)]
ret.append(packed_man)
return ret
else:
return []
elif what == WHATS.CODE:
# 取header里面的app-name, 没给的话就给man
app_name = ctx.request.header('app-name', 'man')
man_id = by_val
code = shortuuid.ShortUUID().random(length=8)
# role是区域经理app端header里面的app-name
key = key_man_code.format(role=app_name, content=code)
# 存放: 二维码(360秒): get <app-name>:code:fe11ad907e2fa779ed2f363ef589d3f9 => 56f1028b2d50c07c80914393
redis_client.setex(key, man_id, 360)
logging.info('Setting Code [%s] => [%s].' % (key, man_id))
return key
elif what == WHATS.PICKUP_LIST:
man = Man.objects(**{by_key: by_val}).only('pick_up_locations').first()
return man.pick_up_locations
elif what == WHATS.SIGN_IN_COUNT:
# 获取人员名字
man = Man.objects(id=by_val).only('name').first()
name = man.name if man.name else ''
# 获取今天凌晨以后的签到记录次数
now = TimeZone.local_now()
start_time, _end_time = TimeZone.day_range(value=now)
start_time = TimeZone.datetime_to_str(start_time)
sign_in_count = SignIn.objects(**{
by_key: by_val,
'create_time__gte': start_time
}).count()
return dict(name=name,
sign_in_count=sign_in_count,
server_time=TimeZone.datetime_to_str(
now, pattern='%Y-%m-%dT%H:%M:%S'))
def op_data():
logging.info("op_data start...")
man_conn = mongodb_client['profile']['man']
shop_conn = mongodb_client['profile']['shop']
express = mongodb_client['aeolus']['express']
opdata = OpData.objects.order_by("-create_time").first()
start_date = date(2016, 5, 20)
if opdata:
start_date = TimeZone.increment_days(
TimeZone.utc_to_local(opdata.create_time), 1).date()
end_date = TimeZone.local_now().date()
for day in TimeZone.date_xrange(start_date, end_date):
start_time, end_time = TimeZone.day_range(value=day)
logging.info("op data %s", day)
data = {}
# ============= 资金 ==============
data['zj_top_up'], data['zj_pay'] = get_zj_statistics(
day, TimeZone.increment_days(day))
# ============= 订单 ==============
data['dd_count'] = express.find({
"create_time": {
"$gte": start_time,
"$lte": end_time
}
}).count()
data['dd_sj_count'] = express.find({
"times.zj_time": {
"$gte": start_time,
"$lte": end_time
}
}).count()
data['dd_tt_count'] = express.find({
"times.parttime_tt_time": {
"$gte": start_time,
"$lte": end_time
}
}).count()
data['dd_error_count'] = express.find({
"times.yc_time": {
"$gte": start_time,
"$lte": end_time
}
}).count()
# ============= 客户 ==============
data['sh_register'] = shop_conn.find({
"create_time": {
"$gte": start_time,
"$lte": end_time
}
}).count()
data['sh_order'] = 0
result = express.aggregate([{
"$match": {
"create_time": {
"$gte": start_time,
"$lte": end_time
}
}
}, {
"$group": {
"_id": "$creator.id",
}
}, {
"$group": {
"_id": "sum",
"count": {
"$sum": 1
}
}
}])
for doc in result:
data['sh_order'] += doc['count']
# ============== 人力 ===============
data['hr_on_job'] = man_conn.find({
"job_description": "parttime",
"status": "STATUS_WORKING"
}).count()
data['hr_active'] = 0
# 在规定时间内收件的人
result = express.aggregate([{
"$match": {
"times.sj_time": {
"$gte": start_time,
"$lte": end_time
}
}
}, {
"$project": {
"man": {
"$arrayElemAt": ["$watchers", 0]
}
}
}, {
"$group": {
"_id": "$man.id"
}
}])
active_list = [_['_id'] for _ in result]
result = express.aggregate([{
"$match": {
"times.parttime_tt_time": {
"$gte": start_time,
"$lte": end_time
}
}
}, {
"$group": {
"_id": "$assignee.id",
}
}])
# 在规定时间内妥投的人
active_list2 = [_['_id'] for _ in result]
temp = set(active_list + active_list2)
data['hr_active'] = len(temp)
# 司机出勤
man_list = man_conn.find({"tel": {"$in": driver_list}})
node_conn = psycopg2.connect(cursor_factory=psycopg2.extras.DictCursor,
database="tlbs",
**CONFIG_POSTGRESQL)
cursor = node_conn.cursor()
data['cl_depart'] = 0
man_id_list = [str(_['_id']) for _ in man_list]
for man_id in man_id_list:
local = arrow.get(day, "local")
cursor.execute(
"""SELECT count(*)
from trans.bs_trans_deliver_sign_record
where deliver_id = %s and create_time >= %s and create_time <= %s
""", (man_id, local.floor("day").strftime(TIME_PATTERN),
local.ceil("day").strftime(TIME_PATTERN)))
doc = cursor.fetchone()
if doc and int(doc[0]) > 0:
data['cl_depart'] += 1
cursor.execute(
"""
SELECT count(*)
FROM trans.bs_trans_bind_node_info
WHERE deliver_id = ANY(%s)
""", (man_id_list, ))
doc = cursor.fetchone()
if not doc:
data['cl_shift'] = 0
else:
data['cl_shift'] = int(doc[0])
# =======================================
create_time = TimeZone.datetime_to_utc(
TimeZone.naive_to_aware(datetime(day.year, day.month, day.day, 1)))
data["create_time"] = create_time
OpData(**data).save()
logging.info("op_data end...")
def shop_reward():
logging.info("invitation shop reward start...")
man_conn = mongodb_client['profile']['man']
shop_conn = mongodb_client['profile']['shop']
express = mongodb_client['aeolus']['express']
now = TimeZone.utc_now()
local = TimeZone.local_now()
cursor = shop_conn.find({
"recommended_by.tel": {
"$exists": True
},
"recommended_by.time": {
"$gt": TimeZone.decrement_months(now, 2)
}
})
m_type = "shop_invitation"
for doc in cursor:
man = man_conn.find_one({"tel": doc['recommended_by']['tel']})
if not man:
logging.warning("no man tel:%s", doc['recommended_by']['tel'])
continue
shop_id = str(doc['_id'])
shop_name = utf8(doc.get("name", "新客户"))
man = format_man(man, 'parttime')
invite_time = TimeZone.naive_to_aware(doc['recommended_by']['time'],
pytz.utc)
# 拿出这个人对应商户所有的奖励, 用来计算已经奖励了多少钱
reward_list = Rewards.objects(man__id=man['id'],
source__shop_id=shop_id,
m_type=m_type)
# 计算还剩多少可以奖励
count = 0
for r in reward_list:
count += r.money
count = int(count)
left = 100 - count
# 如果没剩了, 不能再奖励了
if left <= 0:
continue
# 计算最近一次需要计算的日期,
try:
reward = reward_list[0]
start_date = TimeZone.utc_to_local(
TimeZone.increment_days(reward.create_time, 1)).date()
except IndexError:
start_date = TimeZone.utc_to_local(invite_time).date()
# 最多计算到昨天
end_date = local.date()
# 从开始日期开始到昨天都需要计算
for day in TimeZone.date_xrange(start_date, end_date):
start_time, end_time = TimeZone.day_range(value=day)
# 限制一下start_time
start_time = start_time if start_time > invite_time else invite_time
_count = express.find({
"creator.id": shop_id,
"times.sj_time": {
"$gte": start_time,
"$lte": end_time
},
}).count()
if _count == 0:
continue
c = _count if _count < left else left
money = c * 1.0
# 先将时间加上本地tzinfo, 然后再转换时区, 直接指定tzinfo会出错, 差6分钟, 故意躲加1小时
create_time = TimeZone.datetime_to_utc(
TimeZone.naive_to_aware(
datetime(day.year, day.month, day.day, 1)))
Rewards(m_type=m_type,
title="邀请客户",
desc="%s当日下单%s单" % (shop_name, c),
man=man,
money=money,
source={
"shop_id": shop_id,
"shop_name": shop_name
},
create_time=create_time).save()
logging.info(
"reward: m_type[%s], date[%s], man[%s], shop_id[%s], money[%s]",
m_type, day, man['id'], shop_id, money)
left -= c
if left <= 0:
break
for data_record in data:
try:
baidu_push_v3(channel_id=data_record['channel_id'],
msg=json.dumps(data_record.get('msg', ''),
ensure_ascii=False),
device_type=PLATFORM_ANDROID)
except:
pass
except:
print((traceback.format_exc()))
finally:
pass
if __name__ == '__main__':
print(('[{current_time}] starting server ...'.format(
current_time=TimeZone.local_now())))
connection = pika.BlockingConnection(
pika.ConnectionParameters(host='localhost'))
channel = connection.channel()
channel.exchange_declare(exchange=EXCHANGE_PUSH_MSG, type=TYPE_FANOUT)
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
channel.queue_bind(exchange=EXCHANGE_PUSH_MSG, queue=queue_name)
channel.basic_consume(callback, queue=queue_name, no_ack=True)
channel.start_consuming()
