Exemplo n.º 1
0
def extend(bizplace_id, months=0):
    """
    months: months to add to today
    """
    member_id = test_data.membership_member_id
    date = today + relativedelta.relativedelta(months=months)
    usages_before = usagelib.usage_collection.find(member_ids=[member_id])
    membershiplib.autoextend(bizplace_id, date.month, date.year)
    usages_after = usagelib.usage_collection.find(member_ids=[member_id])
    return len(usages_after) - len(usages_before)
Exemplo n.º 2
0
import os
import sys

rootdir = os.path.join(os.path.dirname(__file__), "..")
sys.path.insert(0, rootdir)
import be.bootstrap

be.bootstrap.start()
import be.apis.membership as membershiplib
import be.apis.user as userlib

userlib.set_context(env.config.system_username, 0)
membershiplib.autoextend()
env.context.pgcursor.connection.commit()