def generating_acronym():
session = DBSession()
firm_list = session.query(models.Firm)\
.filter(models.Firm.status == 0)\
.all()
goods_list = session.query(models.Goods)\
.filter(models.Goods.status == 0)\
.all()
for firm in firm_list:
firm.name = firm.name
for goods in goods_list:
goods.name = goods.name
session.commit()
session.close()
return "success"
def check_msg_token(phone, code):
s = DBSession()
try:
q = s.query(_VerifyCode).filter_by(phone=phone).one()
except:
return False
#print("[验证短信]验证码验证：",q.code,code,q.count,q.wx_id)
# t = (datetime.datetime.now() - q.create_time)
# if t.days == 0 and t.seconds >= 18000:
if q.count != 1:
return False
if q.code == code:
q.count = 0 # make the code useless
s.commit()
s.close()
return True
return False
def update_code(phone,code):
s = DBSession()
try:
print("[VerifyMsg]VerifyCode",code,"have sent to phone:",phone)
q = s.query(_VerifyCode).filter(_VerifyCode.phone == phone).one()
except:
q = None
if q is not None:
q.code = code
q.create_time = datetime.datetime.now()
q.count = 1
# q.phone = phone
else:
v = _VerifyCode()
v.phone = phone
v.code = code
v.count = 1 # when count = 1,code is usefull
s.add(v)
s.commit()
s.close()
return True
if matcher:
shop_name = matcher.group(1)
else:
raise Exception('{} 文件里没找到有效店名')
# 把意向单拿出来
wish_order = models.WishOrder.get_by_id(session, WISH_ORDER_ID)
if not wish_order:
raise Exception('先建个意向单')
station_id = wish_order.station_id
creator_id = wish_order.creator_id
# 从文件名里找一个 Shop，没有就建一个
shop = session.query(models.Shop) \
.filter(models.Shop.abbreviation == shop_name,
models.Shop.station_id == station_id,
models.Shop.status == 0) \
.first()
if not shop:
print('-------------- {} 店还没有，建一个 --------------'.format(shop_name))
shop = models.Shop(
abbreviation=shop_name,
station_id=station_id,
creator_id=creator_id,
)
session.add(shop)
session.flush()
# 建一个订货单
demand_order = session.query(models.DemandOrder) \
.filter(models.DemandOrder.wish_order_id == WISH_ORDER_ID,
"仓库": 85,
}
if __name__ == '__main__':
if not STATION_ID:
raise Exception('先把代码里的常量填了！')
session = DBSession()
filename = '2018-12-30各店意向单(1).xlsx'
wb = load_workbook(filename)
sheet = wb.active
# 先搞个意向单
wish_order = session.query(models.WishOrder) \
.filter(models.WishOrder.wish_date == WISH_DATE,
models.WishOrder.station_id == STATION_ID) \
.first()
if not wish_order:
wish_order = models.WishOrder(
wish_date=WISH_DATE,
status=2, # 直接是提交状态，要改再说
station_id=STATION_ID,
creator_id=CREATOR_ID,
)
session.add(wish_order)
session.flush()
# 每次直接把之前的意向单商品全删了重建
session.query(models.WishOrderGoods) \
.filter(models.WishOrderGoods.wish_order_id == wish_order.id) \
.update({"status": -1})
