Update app.py

app.py

@@ -1,3 +1,59 @@
+import inspect
+from functools import wraps
+from typing import get_type_hints
+
+
+def strict_types(func):
+    sig = inspect.signature(func)
+    type_hints = get_type_hints(func)
+
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        bound_args = sig.bind(*args, **kwargs)
+        bound_args.apply_defaults()
+
+        # Check argument types
+        for name, value in bound_args.arguments.items():
+            expected_type = type_hints.get(name)
+            if expected_type and not isinstance(value, expected_type):
+                raise TypeError(
+                    f"Argument '{name}' expected {
+                        expected_type.__name__}, got {
+                        type(value).__name__}")
+
+        # Call the function
+        result = func(*args, **kwargs)
+
+        # Check return type
+        expected_return = type_hints.get('return')
+        if expected_return and not isinstance(result, expected_return):
+            raise TypeError(
+                f"Return value expected {
+                    expected_return.__name__}, got {
+                    type(result).__name__}")
+
+        return result
+
+    return wrapper
+# main.repy
+
+# 
+# eshare - Share sites easily
+# Copyright (C) 2025  Rihaan Meher
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+# 
+
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel, conint
 import threading
@@ -7,12 +63,13 @@ app = FastAPI()
 live_names = {}
 expiration_times = {}
 
-class LiveNameRequest(BaseModel):
+class LiveNameRequest(BaseModel) :
     name: str
     url: str
     timeout: conint(ge=60, le=3600)
 
-def remove_expired_name(name: str):
+@strict_types
+def remove_expired_name(name: str) :
     time.sleep(expiration_times[name] - time.time())
     if name in live_names:
         del live_names[name]
@@ -20,8 +77,8 @@ def remove_expired_name(name: str):
         print(f"{name} has expired and is available again!")
 
 @app.post("/set_live_name")
-def set_live_name(data: LiveNameRequest):
-    if data.name in live_names:
+def set_live_name(data: LiveNameRequest) :
+    if data.name in live_names :
         remaining_time = max(0, int((expiration_times[data.name] - time.time()) / 60))
         return {"message": f"Name taken. Available in {remaining_time} min(s)."}
 
@@ -34,8 +91,7 @@ def set_live_name(data: LiveNameRequest):
             "access_url": f"/lookup/{data.name}"}
 
 @app.get("/lookup/name")
-def lookup_name(c: str):
-    if c in live_names:
+def lookup_name(c: str) :
+    if c in live_names :
         return {"url": live_names[c]}
-    
-    raise HTTPException(status_code=404, detail="Live name not found")
+    raise HTTPException(status_code=404, detail="Live name not found")