0.2.3 V Alpha

Instance.py

@@ -9,13 +9,14 @@ from tqdm import tqdm
 from indexer import indexer
 import re
 from tvdb import fetch_and_cache_json
+import logging
 
 CACHE_DIR = os.getenv("CACHE_DIR")
 
 download_progress = {}
 
 class Instance:
-    def __init__(self, id, url, cache_dir, index_file, token, repo):
+    def __init__(self, id, url, cache_dir, index_file, token, repo, load_balancer_api,max_retries=5, initial_delay=1):
         self.version = "0.2.3 V Alpha"
         self.id = id
         self.url = url
@@ -27,6 +28,9 @@ class Instance:
         self.TV_STORE_JSON_PATH = os.path.join(cache_dir, "tv_store.json")
         self.download_threads = {}
         self.file_structure = None
+        self.load_balancer_api = load_balancer_api
+        self.max_retries = max_retries
+        self.initial_delay = initial_delay
 
         # Ensure CACHE_DIR exists
         if not os.path.exists(self.CACHE_DIR):
@@ -49,6 +53,7 @@ class Instance:
 
         # Start prefetching metadata
         thread = Thread(target=self.start_prefetching)
+        self.register_to_load_balancer()
         thread.daemon = True
         thread.start()
 
@@ -58,7 +63,8 @@ class Instance:
         cache_size = self.get_cache_size()
 
         report = {
-            "id": self.id,
+            "instance_id":self.id,
+            "instance_url": self.url,
             "film_store": self.read_json(film_store_path),
             "tv_store": self.read_json(tv_store_path),
             "cache_size": cache_size
@@ -379,4 +385,23 @@ class Instance:
 
     def start_prefetching(self):
         """Start the metadata prefetching in a separate thread."""
-        self.prefetch_metadata()
+        self.prefetch_metadata()
+
+
+    def register_to_load_balancer(self):
+        retries = 0
+        delay = self.initial_delay
+
+        while retries < self.max_retries:
+            result = self.load_balancer_api.register_instance(self.id, self.url)
+            if result:
+                logging.info(f'Successfully registered instance {self.id} to load balancer.')
+                return result
+
+            retries += 1
+            logging.warning(f'Attempt {retries} to register instance {self.id} failed. Retrying in {delay} seconds...')
+            time.sleep(delay)
+            delay *= 2  # Exponential backoff
+
+        logging.error(f'Failed to register instance {self.id} to load balancer after {self.max_retries} attempts.')
+        return None

TODO.md

@@ -1,5 +1,4 @@
 ## TODO
-* Create Instance class
 * add a method to register the instance to Load Balancer on ```[POST]<load_balancer_url>/api/register``` route
 * add `[GET] /api/get/report` route to return a report consist of film_store.json, tv_store.json, cache_size
 * add two api routes to delete a certain film or episode

api.py

@@ -0,0 +1,20 @@
+import requests
+import logging
+
+class LoadBalancerAPI:
+    def __init__(self, base_url):
+        self.base_url = base_url
+
+    def register_instance(self, instance_id, instance_url):
+        data = {
+            "url": instance_url
+        }
+        api_endpoint = f'{self.base_url}/api/register'
+        
+        try:
+            response = requests.post(api_endpoint, data=data)
+            response.raise_for_status()
+            return response.json()  # Assuming the API returns JSON
+        except requests.exceptions.RequestException as e:
+            logging.error(f'Failed to register instance {instance_id} to load balancer: {e}')
+            return None

app.py

@@ -5,6 +5,7 @@ import json
 from threading import Thread
 import urllib.parse
 from Instance import Instance
+from api import LoadBalancerAPI
 
 app = Flask(__name__)
 CORS(app)
@@ -16,11 +17,12 @@ TOKEN = os.getenv("TOKEN")
 REPO = os.getenv("REPO")
 ID = os.getenv("ID")
 URL = os.getenv("URL")
+LOAD_BALANCER_URL = os.getenv("LOAD_BALANCER_URL")
 
-instance = Instance(id=ID, url=URL, cache_dir=CACHE_DIR, index_file=INDEX_FILE, token=TOKEN, repo=REPO)
+load_balancer_api = LoadBalancerAPI(base_url=LOAD_BALANCER_URL)
+instance = Instance(id=ID, url=URL, cache_dir=CACHE_DIR, index_file=INDEX_FILE, token=TOKEN, repo=REPO, load_balancer_api=load_balancer_api)
 
 # API Endpoints
-
 @app.route('/api/film/<title>', methods=['GET'])
 def get_movie_api(title):
     """Endpoint to get the movie by title."""