def get_data():
try:
data = storage.load_user_all()
except Exception:
storage.new_storage()
data = storage.load_user_all()
return data
def user_profile_setemail(token, email):
data = storage.load_user_all()
helper.check_email(email)
helper.check_email_exist(email, data)
user_all = storage.load_user_all()
u_id = helper.get_id(token, data)
user_all[str(u_id)]['email'] = email
storage.save_user_all(user_all)
def user_profile_sethandle(token, handle_str):
# InputError if length of handle_str invalid
user_all = storage.load_user_all()
if len(handle_str) < 2 or len(handle_str) > 20:
raise error.InputError
# InputError if handle is taken by another user
u_id = str(helper.get_id(token, user_all))
for user in user_all:
if handle_str == user_all[user]['handle']:
raise error.InputError
user_all = storage.load_user_all()
user_all[u_id]['handle'] = handle_str
storage.save_user_all(user_all)
def user_profile_setname(token, name_first, name_last):
helper.check_name(name_first, name_last)
user_all = storage.load_user_all()
u_id = helper.get_id(token, user_all)
user_all[str(u_id)]['name_first'] = name_first
user_all[str(u_id)]['name_last'] = name_last
storage.save_user_all(user_all)
def test_check_argumetns():
reset()
data = register_user()['data']
data_send = storage.load_user_all()
name_first = list(data_send.values())[0]['name_first']
name_last = list(data_send.values())[0]['name_last']
assert data['name_first'] == name_first
assert data['name_last'] == name_last
def admin_userpermission_change(token,u_id,permission_id):
user_all = storage.load_user_all()
if u_id not in user_all:
raise InputError()
user = user_all[u_id]
# if not valid_perm_id(permission_id)
user['permission_id'] = permission_id
storage.save_user_all(user_all)
return {}
def user_remove(token, u_id):
# Check valid user
user_all = storage.load_user_all()
helper.check_user(u_id, user_all)
# The authorised user is not an owner of the slackr
t_u_id = str(helper.get_id(token, user_all))
helper.check_user(t_u_id, user_all)
delete = [i for i in user_all if i['u_id'] == u_id]
user_all[str(u_id)].remove(delete[0])
storage.save_user_all(user_all)
def channel_detail(token, channel_id):
channel_data = get_data()
data = storage.load_user_all()
helper.check_channel(channel_id, channel_data)
u_id = helper.get_id(token,data)
helper.check_access(u_id,channel_data, channel_id)
return {
"name":channel_data[channel_id]['name'],
"owner":channel_data[channel_id]['owner'],
"members": channel_data[channel_id]['member']
}
def test_create_arguments():
reset()
channel_property = create_channel()['channel_property']
user_data = storage.load_user_all()
channel_data = storage.load_channel_all()
channel_name = list(channel_data.values())[0]['name']
is_public = list(channel_data.values())[0]['access']
u_id = list(user_data.values())[0]['u_id']
owner_uid = get_id(channel_property['token'],user_data)
assert channel_name == channel_property['name']
assert owner_uid == u_id
assert is_public == channel_property['is_public']
def users_all(token):
user_all = storage.load_user_all()
users = {"users": []}
for user in user_all:
user_data = {
'u_id': user_all[user]['u_id'],
'email': user_all[user]['email'],
'name_first': user_all[user]['name_first'],
'name_last': user_all[user]['name_last'],
'handle_str': user_all[user]['handle'],
'profile_img_url': user_all[user]['profile_img_url'],
}
users["users"].append(user_data)
return users
def user_profile(token, u_id):
user_all = storage.load_user_all()
# helper.check_user(u_id, user_all)
user = {
'u_id': u_id,
'email': user_all[str(u_id)]['email'],
'name_first': user_all[str(u_id)]['name_first'],
'name_last': user_all[str(u_id)]['name_last'],
'handle_str': user_all[str(u_id)]['token'],
'profile_img_url': user_all[str(u_id)]['profile_img_url'],
}
return user
def auth_passwordreset(code,new_pass):
# check if the new email is valid
new_pass = encrypt_pass(new_pass)
user = storage.load_user_all()
# check if the new email is not already in use
data = get_data()
# check if the code is valid
if code not in data: raise error.InputError
email = data[code]
# find the user to change their email
i = [i for i in user if user[i]['email'] == email][0]
user[i]['encrypted_password'] = new_pass
del(data[code])
storage.save_code_file(data)
storage.save_user_all(user)
# auth_passwordreset_request("*****@*****.**")
# auth_passwordreset("0312138261","*****@*****.**")
def user_profiles_uploadphoto(token, img_url, x_start, y_start, x_end, y_end):
# get file from img_url.
response = requests.get(img_url, stream=True)
if response.status_code != 200:
raise error.InputError("Not 200 code!")
# make a new file with filename: image12characters.jpg.
filename = f'{get_random_alphaNumeric_string(12)}.jpg'
# stored in filepath: /pictures.
#filepath = f'/Users/uni/Documents/Uni/20T1/Comp1531/Assignments/iteration_1/W17A-17/src/static/pictures/{filename}'
filepath = f'{os.getcwd()}/static/pictures/{filename}'
with open(filepath, 'wb') as OUT_FILE:
# copy data into file without checking its type.
shutil.copyfileobj(response.raw, OUT_FILE)
del response
# check file is a jpg.
try:
with Image.open(filepath) as IMAGE_FILE:
if IMAGE_FILE.format != 'JPEG':
# remove the file we created.
os.remove(filepath)
raise InputError("Image given is not a jpeg.")
except IOError:
# remove the file we created.
os.remove(filepath)
raise InputError(
"PIL failed to open the file. Likely file isn't a valid image.")
x_start = int(x_start)
x_end = int(x_end)
y_start = int(y_start)
y_end = int(y_end)
# we now know filepath *is* a jpg image.
# check the dimensions of the image.
with Image.open(filepath) as IMAGE_FILE:
# IMAGE_FILE.size returns tuple (width, height) of IMAGE_FILE.
width, height = IMAGE_FILE.size
if ((x_end - x_start) >= width) or ((y_end - y_start) >= height):
# remove the file we created.
os.remove(filepath)
raise InputError(
"You gave the wrong dimensions for cropping dummy.")
# crop image.
with Image.open(filepath) as IMAGE_FILE:
box = (x_start, y_start, x_end, y_end)
CROPPED_IMAGE_FILE = IMAGE_FILE.crop(box)
# save cropped image into a new file.
new_filename = 'cropped' + filename
new_filepath = f'{os.getcwd()}/static/pictures/{new_filename}'
CROPPED_IMAGE_FILE.save(new_filepath)
# for debugging purposes, don't bother removing the IMAGE_FILE.
### hmmm now how to let front end know where the file is...
user_all = storage.load_user_all()
u_id = helper.get_id(token, user_all)
profile_img_url = 'http://localhost:8050/static/pictures/' + new_filename
user_all[str(u_id)]['profile_img_url'] = profile_img_url
storage.save_user_all(user_all)
# update profile_img_url for the user in all other channels.
channel_all = storage.load_channel_all()
# woops didn't know channel was the key, not the value.
for channel_id in channel_all:
channel = channel_all[channel_id]
owner_list = channel['owner']
for owner in owner_list:
if owner['u_id'] == u_id:
owner['profile_img_url'] = profile_img_url
member_list = channel['member']
for member in member_list:
if member['u_id'] == u_id:
member['profile_img_url'] = profile_img_url
storage.save_channel_all(channel_all)
return {}
