diff --git a/main.py b/main.py index 9381cf9..cfd24e0 100644 --- a/main.py +++ b/main.py @@ -81,16 +81,17 @@ def edit(): avatar=account.get_avatar(domain) host = request.host - + admin=False if domain.lower() == ADMIN_DOMAIN: links = db.get_all_links() + admin=True else: links = db.get_users_links(domain) link_count = len(links) if links == False: links = "

No links created yet

" else: - links = render.links(links,host) + links = render.links(links,host,admin) return render_template('dash.html',domain=domain,avatar=avatar,host=host,links=links,link_count=link_count) @app.route('/dash', methods=['POST']) diff --git a/render.py b/render.py index 0dd2781..9a74eaf 100644 --- a/render.py +++ b/render.py @@ -1,9 +1,11 @@ -def links(links,host=""): +def links(links,host="",show_domain=False): host = "https://"+host+"/" html = "" for link in links: html += "\n" return html \ No newline at end of file