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)
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()