def count_increate(rec, cat_id, num):
entry = TabUsage.update(
timestamp=int(time.time()),
count=num + 1,
tag_id=cat_id,
).where(TabUsage.uid == rec)
entry.execute()
def add_or_update(user_id, post_id, kind):
'''
Create the record if new, else update it.
'''
rec = MUsage.query_by_signature(user_id, post_id)
cate_rec = MInfor2Catalog.get_first_category(post_id)
if cate_rec:
cat_id = cate_rec.tag_id
else:
return False
if rec.count() > 0:
logger.info('Usage update: {uid}'.format(uid=post_id))
rec = rec.get()
query = TabUsage.update(kind=kind).where(TabUsage.uid == rec.uid)
query.execute()
MUsage.count_increate(rec.uid, cat_id, rec.count)
else:
logger.info('Usage create: {uid}'.format(uid=post_id))
TabUsage.create(
uid=tools.get_uuid(),
post_id=post_id,
user_id=user_id,
count=1,
tag_id=cat_id,
timestamp=int(time.time()),
kind=kind,
)
def update_field(uid, post_id=None):
if post_id:
entry = TabUsage.update(
post_id=post_id
).where(TabUsage.uid == uid)
entry.execute()
