def add_to_app(app_name, app_url):
app = get_app(name=app_name).fetchone()
if app:
return create_app_url(app_id=app[0], domain=app_url).fetchall()
else:
app = create_app(name=app_name).fetchone()
return create_app_url(app_id=app[0], domain=app_url).fetchall()
def remove_to_app(app_name, app_url):
app = get_app(name=app_name).fetchone()
if app:
delete_app_url(app_id=app[0], domain=app_url)
return get_app_url(app_id=app[0]).fetchall()
else:
return []
def check_urls(url, user_urls):
if len(user_urls) == 0:
return (False,0,0)
elif user_urls[0][2] in url:
if len(user_urls[0]) == 3: #its app
return (True, user_urls[0][2], get_app(id=user_urls[0][1]).fetchone()[1])
elif len(user_urls[0]) == 4: #its url
return (True, user_urls[0][2], '_')
else:
return check_urls(url, user_urls[1:])
def unblock(user_ip, domain=None, app_name=None):
if app_name != None:
user = get_user(ip=user_ip).fetchone()
app = get_app(name=app_name).fetchone()
if user and app:
return delete_block(user_id=user[0], app_id=app[0])
elif user and not app:
print("App not exist")
else:
return f"User {user_ip} has no locks."
elif domain != None:
user = get_user(ip=user_ip).fetchone()
return delete_block(user_id=user[0], domain=domain)
def block(user_ip, domain=None, app_name=None):
if app_name != None:
user = get_user(ip=user_ip).fetchone()
app = get_app(name=app_name).fetchone()
if user and app:
return create_block(user_id=user[0], app_id=app[0])
elif user and not app:
app = create_app(name=app_name).fetchone()
return create_block(user_id=user[0], app_id=app[0])
elif not user and app:
user = create_user(user_ip).fetchone()
return reate_block(user_id=user[0], app_id=app[0])
else:
user = create_user(user_ip).fetchone()
app = create_app(name=app_name).fetchone()
return create_block(user_id=user[0], app_id=app[0])
elif domain != None:
user = get_user(ip=user_ip).fetchone()
if not user:
user = create_user(user_ip).fetchone()
return create_block(user_id=user[0], domain=domain)
def get_to_app(app_name):
app = get_app(name=app_name).fetchone()
if app:
return get_app_url(app_id=app[0]).fetchall()
else:
return []