Creation

1 files changed, 74 insertions, 0 deletions

diff --git a/app/func/log.py b/app/func/log.py
new file mode 100644
index 0000000..0a7dddc
--- /dev/null
+++ b/app/func/log.py
@@ -0,0 +1,74 @@
+import ip2locationio
+import sqlalchemy
+import datetime
+import validators
+from ua_parser import user_agent_parser
+from dotenv import load_dotenv
+import os
+from ip2locationio.ipgeolocation import IP2LocationIOAPIError
+
+from db import engine
+
+load_dotenv()
+try:
+    ip_to_location = os.getenv('IP_TO_LOCATION').upper().replace('"', '')
+    if ip_to_location == 'TRUE':
+        api_key = os.getenv('API_KEY').replace('"', '')
+    else:
+        api_key = "NO_API_KEY"
+
+    base_url = os.getenv('BASE_URL').replace('"', '')
+# .env File does not exist - likely a docker run
+except AttributeError:
+    ip_to_location = str(os.environ['IP_TO_LOCATION']).upper().replace('"', '')
+    if ip_to_location == 'TRUE':
+        api_key = str(os.environ('API_KEY')).replace('"', '')
+    else:
+        api_key = "NO_API_KEY"
+
+    base_url = str(os.environ('BASE_URL')).replace('"', '')
+
+if not validators.url(base_url):
+    print(base_url)
+    print('BASE_URL varaible is malformed.')
+    exit()
+
+configuration = ip2locationio.Configuration(api_key)
+ipgeolocation = ip2locationio.IPGeolocation(configuration)
+
+"""
+Create a new log record whenever a link is visited
+"""
+def log(link, request):
+    with engine.begin() as conn:
+        try:
+            redirect_link, owner = conn.execute(sqlalchemy.text('SELECT redirect_link, owner FROM links WHERE link = :link'), [{'link': link}]).fetchone()
+        except TypeError:
+            return base_url
+
+    with engine.begin() as conn:
+        if ip_to_location == 'TRUE':
+            # Get IP to GEO via IP2Location.io
+            try:
+                data = ipgeolocation.lookup(request.remote_addr)
+                location = f'{data["country_name"]}, {data["city_name"]}'
+                isp = data['as']
+            # Fatal error, API key is invalid or out of requests, quit
+            except IP2LocationIOAPIError:
+                print('Invalid API key or insifficient credit. Change .env file if you do not need IP to location feature.')
+                location = '-, -'
+                isp = '-'
+        else:
+            location = '-, -'
+            isp = '-'
+
+        timestamp = datetime.datetime.now()
+        ip = request.remote_addr
+        user_agent = request.user_agent.string
+        ua_string = user_agent_parser.Parse(user_agent)
+        browser = ua_string['user_agent']['family']
+        os = f'{ua_string["os"]["family"]} {ua_string["os"]["major"]}'
+
+        conn.execute(sqlalchemy.text('INSERT INTO records (owner, link, timestamp, ip, location, browser, os, user_agent, isp) VALUES (:owner, :link, :timestamp, :ip, :location, :browser, :os, :user_agent, :isp)'), [{'owner': owner, 'link': link, 'timestamp': timestamp, 'ip': ip, 'location': location, 'browser': browser, 'os': os, 'user_agent': user_agent, 'isp': isp}])
+
+    return redirect_link
\ No newline at end of file