db.py 文件源码

def set_live_subscription(subscriptor, member_ids):
    """
    ?????????????????????????
    subscriptor??????????tuple?
    ??????????????????????
    """
    if not isinstance(member_ids, (list, tuple)):
        member_ids = [member_ids]
    if isinstance(subscriptor, (list, tuple)) and len(subscriptor) == 2:
        params = {
            'pocket48_phonenum': subscriptor[0],
            'pocket48_password': subscriptor[1],
        }
    elif isinstance(subscriptor, (str, int)):
        params = {'pocket48_phonenum': subscriptor}
    else:
        return False

    try:
        s, _ = Subscriptor.get_or_create(**params)
        for mid in member_ids:
            LiveSubscription.get_or_create(
                    subscriptor=s, member=Member.get(member_id=mid))
    except peewee.OperationalError:
        return False
    return True