def reset_supplier_declaration(stage, framework_slug, reason, email, supplier_id):
data_api_token = get_auth_token('api', stage) if stage != 'development' else 'myToken'
data_api_client = DataAPIClient(get_api_endpoint_from_stage(stage), data_api_token)
if email:
user = data_api_client.get_user(email_address=email)
if not user:
print(f'No user found for email address `{email}`')
exit(1)
user_supplier_id = user['users']['supplier']['supplierId']
if user_supplier_id and supplier_id and user_supplier_id != supplier_id:
print(f'Email address provided does not match with supplier provided. Email address `{email}` is '
f'associated with supplierId `{supplier_id}`. Script was called with supplierId `{supplier_id}`.')
exit(2)
supplier_id = user_supplier_id
try:
supplier_framework = data_api_client.get_supplier_framework_info(supplier_id=supplier_id,
framework_slug=framework_slug)
print(f"Current supplier declaration: {supplier_framework['frameworkInterest']['declaration']}")
except HTTPError:
print(f'No supplier framework found for supplierId `{supplier_id}` on framework `{framework_slug}`.')
exit(3)
if not supplier_framework:
print(f'No supplier framework/interest record found for supplierId `{supplier_id}` on framework '
f'`{framework_slug}`.')
exit(4)
data_api_client.set_supplier_declaration(supplier_id=supplier_id, framework_slug=framework_slug, declaration={},
user=f'{getpass.getuser()} - {reason}')
data_api_client.set_supplier_framework_prefill_declaration(supplier_id=supplier_id, framework_slug=framework_slug,
prefill_declaration_from_framework_slug=None,
user=f'{getpass.getuser()} - {reason}')
print(f'Supplier declaration for supplierId `{supplier_id}` on framework `{framework_slug}` has been reset.')
import sys
sys.path.insert(0, '.')
from os import environ
from sys import exit
from dmapiclient import DataAPIClient
from dmscripts.helpers.auth_helpers import get_auth_token
from dmutils.env_helpers import get_api_endpoint_from_stage
if __name__ == "__main__":
data_api_client = DataAPIClient(
get_api_endpoint_from_stage(environ["STAGE"].lower()),
get_auth_token('api', environ["STAGE"].lower()),
)
email_address = environ["ACCOUNT_EMAIL"]
user = data_api_client.get_user(email_address=email_address)
if not user:
print(f"User {email_address!r} not found")
exit(2)
if not data_api_client.update_user_password(
user["users"]["id"],
environ["ACCOUNT_PASSWORD"],
"set-dm-password-by-email.py",
):
print(f"Failed to set password for {email_address!r}")
exit(3)
