def put(self): """PUT endpoint that register a new user. Returns dict Dictionary containing error or success key. int Response Code """ token = request.headers.get("authorization-code") if token is not None: data = request.get_json(force=True) id = data.get("user_id", None) name = data.get("user_name", None) if id is None or name is None: return ( { "error": "Some body arguments is missing. It needs id, email and name" }, 463, ) user = User(id=id, name=name, query_id=id) auxapi = ContactApi() contacts = auxapi._get_list_of_contacts(grouped=False) if contacts[1] == 200: contacts_statistics = Contact.get_quantity_per_domain( contacts[0]) contacts_statistics_organization = Contact.get_quantity_per_organization( contacts[0]) contacts_statistics_jobtitle = Contact.get_quantity_per_job( contacts[0]) contacts_statistics_city = Contact.get_quantity_per_city( contacts[0]) contacts_statistics_region = Contact.get_quantity_per_region( contacts[0]) user.contacts_statistics = { "domain": contacts_statistics["contacts"], "organization": contacts_statistics_organization["contacts"], "jobtitle": contacts_statistics_jobtitle["contacts"], "city": contacts_statistics_city["contacts"], "region": contacts_statistics_region["contacts"], } user.save() return {"success": "User Saved"}, 200 return contacts else: return NO_AUTH_CODE
def create_test_user() -> dict: """Create a test fuel supplier user""" user = User() generated_name = 'test_{0}'.format(str(uuid.uuid4())[8:]) user.username = generated_name user.first_name = 'Test' user.last_name = 'User' user.display_name = 'Test User' user.organization = Organization.objects.get_by_natural_key( DataCreationUtilities.create_test_organization()['name']) user.save() user.refresh_from_db() return {'id': user.id, 'username': user.username}
def create(self, validated_data): password = validated_data.pop('password') user = User(**validated_data) user.set_password(password) user.save() return user