worker: Added initial files

worker/main.py

@@ -0,0 +1,58 @@
+# API server
+
+from flask import Flask, request, jsonify
+import dotenv
+import os
+
+dotenv.load_dotenv()
+
+
+app = Flask(__name__)
+
+# API route for POST requests for new-site (takes variable 'domain')
+@app.route('/new-site', methods=['POST'])
+def new_site():
+    # Get the URL from the request
+    domain = request.args.get('domain')
+    count = get_sites_count()
+
+    if site_exists(domain):
+        return jsonify({'error': 'Site already exists', 'success': 'false'})
+
+    # Add site to file
+    sites_file = open('sites.txt', 'a')
+    sites_file.write(domain + '\n')
+
+    # Return the domain and the number of sites
+    return jsonify({'domain': domain, 'count': count})
+
+def get_sites_count():
+    # If file doesn't exist, create it
+    try:
+        sites_file = open('sites.txt', 'r')
+    except FileNotFoundError:
+        sites_file = open('sites.txt', 'w')
+        sites_file.close()
+        sites_file = open('sites.txt', 'r')
+
+    # Return number of lines in file
+    return len(sites_file.readlines())
+
+def site_exists(domain):
+    # If file doesn't exist, create it
+    try:
+        sites_file = open('sites.txt', 'r')
+    except FileNotFoundError:
+        sites_file = open('sites.txt', 'w')
+        sites_file.close()
+        sites_file = open('sites.txt', 'r')
+
+    # Check if domain is in file
+    if domain in sites_file.read():
+        return True
+    else:
+        return False
+
+# Start the server
+if __name__ == '__main__':
+    app.run(debug=False, port=5000)