def create_users():
user1 = User(email="*****@*****.**",
username='******',
password=User.generate_hash('helloworld'),
isVerified=True).create()
user2 = User(email="*****@*****.**",
username='******',
password=User.generate_hash('helloworld')).create()
def create_users():
user1 = User(username='******', email="*****@*****.**", \
password=User.generate_hash('hello'), isVerified=True).create()
user2 = User(username='******', email="*****@*****.**", \
password=User.generate_hash('hello')).create()
user3 = User(username='******', email= "*****@*****.**", \
password=User.generate_hash('hello'), isVerified=True, role ="admin").create()
def create_user():
user1 = User(
username='******',
email='*****@*****.**',
password=User.generate_hash('nghi!abc123'),
is_verified=True,
).create()
user2 = User(
username='******',
email='*****@*****.**',
password=User.generate_hash('nghi!abc123'),
is_verified=False,
).create()
def update_upc_map(self, request: Request):
user = get_user_id_from_request(request)
map_list = json.loads(request.data.get('map_list', None))
if user and map_list:
user = User(id=user)
for code in map_list.get('upc_map', {}):
upc_id = Upc.get_upc_id(code.get('upc__upc', None))
item_id = Item.get_item_id(code.get('item__item_code', None))
if UpcMap.objects.filter(upc=upc_id,
user_id=user,
item=item_id):
to_update = UpcMap.objects.get(upc=upc_id,
user_id=user,
item=item_id)
to_update.amount = code.get('amount')
to_update.save()
elif not UpcMap.objects.filter(
upc=upc_id, user_id=user, item=item_id):
upc_list = UpcMap(user_id=user,
upc=upc_id,
item=item_id,
amount=code.get('amount', 0))
upc_list.save(force_insert=True)
return Response(status=200)
else:
return HttpResponse(status=401)
def set_initial_inventory(self, request: Request):
user = get_user_id_from_request(request)
new_inventory = json.loads(request.data.get('inventory', None))
if user and new_inventory:
user = User(id=user)
for item in new_inventory:
item_id = Item.get_item_id(item.get('item', None))
item_id = Item(id=item_id.id)
if Supply.objects.filter(user_id=user, item_id=item_id):
old_amount = Supply.objects.get(user_id=user,
item_id=item_id)
new_amount = old_amount.amount + int(item.get('amount', 0))
to_update = Supply.objects.get(user_id=user,
item_id=item_id)
to_update.amount = new_amount
to_update.save()
else:
inventory_item = Supply.objects.create(user_id=user,
item_id=item_id,
amount=item.get(
'amount', 1),
date=datetime.now())
return Response(status=200)
else:
return HttpResponse(status=401)
def signup(username, password):
if db.session.query(exists().where(User.username == username)).scalar():
raise HTTPNotFound(title='Cannot', description='asdfas')
u = User(username=username, password=password)
print(u)
db.session.add(u)
return {'token': generate_token(u)}
def create(self, validated_data: dict) -> User:
"""
Create a new bulk quote entity
:param validated_data: data to create with
:return: Created Entity
"""
instance = User(**validated_data)
instance.save()
return instance
def create_portal(self, request: Request):
portal = request.data.get('portal', None)
if portal:
new_user = User(portal_firestore_key=portal)
new_user.save()
return Response(status=200)
else:
return HttpResponse(status=401)
def create_user_in_db(db: Session, user: UserCreate):
db_user = User(
email=user.email,
password=User.generate_hash(user.password),
username=user.email,
role=user.role,
)
db.add(db_user)
db.commit()
db.refresh(db_user)
return db_user
def new_user():
username = request.json.get('username')
password = request.json.get('password')
if username is None or password is None:
abort(400) # missing arguments
if User.query.filter_by(username=username).first() is not None:
abort(400) # existing user
user = User(username=username)
user.hash_password(password)
db.session.add(user)
db.session.commit()
return (jsonify({'username': user.username}))
def create_inventory(self, request: Request):
user = get_user_id_from_request(request)
if user:
item = Item(id=0)
user = User(id=user)
new_inventory = Supply(user_id=user,
item_id=item,
amount=0,
date=datetime.datetime.now())
new_inventory.save()
return Response(status=200)
else:
return HttpResponse(status=401)
def submit_upc_map(self, request: Request):
user = get_user_id_from_request(request)
map_list = json.loads(request.data.get('map_list', None))
if user and map_list:
user = User(id=user)
for code in map_list:
for list_item in code:
upc_id = Upc.get_upc_id(list_item.get('upc', None))
upc_id = Upc(id=upc_id.id)
item_id = Item.get_item_id(list_item.get('item', None))
item_id = Item(id=item_id.id)
upc_list = UpcMap(user_id=user,
upc=upc_id,
item=item_id,
amount=list_item.get('amount', 0))
upc_list.save(force_insert=True)
return Response(status=200)
else:
return HttpResponse(status=401)
def submit_upc(self, request: Request):
user = get_user_id_from_request(request)
code_list = json.loads(request.data.get('upc_list', None))
if user and code_list:
user = User(id=user)
for code in code_list:
upc_id = Upc.get_upc_id(code.get('upc', None))
upc_id = Upc(id=upc_id.id)
for i in range(1, int(code.get('amount', None)) + 1):
upc_list = UpcList(user_id=user,
upc_id=upc_id,
date=datetime.now())
supply = Supply()
update_supply = supply.update_supply(user=user,
upc_id=upc_id.id)
if not update_supply:
raise ValueError(
'The submitted UPC has not been mapped for this practice.'
)
upc_list.save(force_insert=True)
return Response(status=200)
else:
return HttpResponse(status=401)
def signup():
"""User Signup
POST DATA:
fname : First Name of the User
lname : Last Name of the User
email : Email of the User
password : Password of the user
Returns:
201 -- success
202 -- fail (user already exists)
401 -- fail (unknown error)
"""
data = request.get_json(silent=True)
fname, lname, email, password, blood, dob, weight, lat, lon, phone = (
data.get("firstName"),
data.get("lastName"),
data.get("email"),
data.get("password"),
data.get("blood"),
data.get("dob"),
data.get("weight"),
data.get("lat"),
data.get("long"),
data.get("phone"),
)
user = User.query.filter_by(email=email).first()
if not user:
try:
# database ORM object
user = User(
first_name=fname,
last_name=lname,
email=email,
phone=phone,
password=password,
location=(lat, lon),
bloodgroup=blood,
dob=dob,
weight=weight,
)
# insert user
db.session.add(user)
db.session.commit()
responseObject = {
"status": "success",
"message": "Successfully registered.",
}
return make_response(jsonify(responseObject), 201)
except Exception:
responseObject = {
"satus": "fail",
"message": "Some error occured. Please try again.",
}
return make_response(jsonify(responseObject), 401)
else:
responseObject = {
"status": "fail",
"message": "User already exists. Please Log in.",
}
return make_response(jsonify(responseObject), 202)
from flask_restplus import Resource
from flask import jsonify, request
from flask_jwt_extended import (create_access_token)
# Local Imports
from api.helpers.schema.user import (UserSignUpSchema, UserLoginSchema)
from api.models.users import User
user = User()
class RegisterUserResource(Resource):
def post(self):
request_data = request.get_json()
new_user_data = UserSignUpSchema().load_json_data(request_data)
new_user_data.pop('confirm_password')
user.save(**new_user_data)
response = jsonify({'message': 'user created successfully'})
response.status_code = 201
return response
class LoginUserResource(Resource):
def post(self):
user_credentials = request.get_json()
logged_user = UserLoginSchema().load_json_data(user_credentials)
logged_user_data = {
'username': logged_user['username']
} if 'username' in user_credentials else { # noqa E501
'email': logged_user['email']
def admin_access_token(client, test_admin, test_admin_login):
User().save(**test_admin)
response = client.post('/api/v1/auth/login',
data=test_admin_login,
headers={'content-type': 'application/json'})
return json.loads(response.data)['access_token']