Initial Commit

DBFiles/___init__.py

@@ -0,0 +1,7 @@
+
+from .database import PUBLICFilesDB
+from FileStream.config import Telegram
+
+DB_Controller = PUBLICFilesDB(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
+
+

DBFiles/database.py

@@ -0,0 +1,78 @@
+import re
+import time
+import pymongo
+import motor.motor_asyncio
+from bson.objectid import ObjectId
+from bson.errors import InvalidId
+from bson.json_util import dumps
+
+from pymongo.errors import PyMongoError
+
+class PUBLICFilesDB:
+    def __init__(self, uri, database_name):
+        self._client = motor.motor_asyncio.AsyncIOMotorClient(uri)
+        self.db = self._client[database_name]
+        self.files = self.db.Public_Files
+
+    async def GetSeriesAllBy10(self, offset=None):
+        try:
+            filter = {"type": "TVSeries"}  # Define the filter for movies
+            total_results = await self.files.count_documents(filter)
+            offset=int( offset if offset else 0)
+            next_offset = offset + 10
+            if next_offset >= total_results:  # Changed > to >= to handle the last page correctly
+                next_offset = ''
+            cursor = self.files.find(filter)  # Apply the filter
+            cursor.sort('$natural', pymongo.DESCENDING)  # Use pymongo.DESCENDING
+            cursor.skip(offset).limit(10)
+            files = await cursor.to_list(length=10)
+            return files, next_offset
+
+        except PyMongoError as e:
+            print(f"MongoDB error: {e}")
+            return [], ''  # Return empty list and '' on error
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            return [], ''
+
+    async def GetMoviesAllBy10(self, offset=None):
+        try:
+            filter = {"type": "Movie"}  # Define the filter for movies
+            total_results = await self.files.count_documents(filter)
+            offset=int( offset if offset else 0)
+            next_offset = offset + 10
+            if next_offset >= total_results:  # Changed > to >= to handle the last page correctly
+                next_offset = ''
+            cursor = self.files.find(filter)  # Apply the filter
+            cursor.sort('$natural', pymongo.DESCENDING)  # Use pymongo.DESCENDING
+            cursor.skip(offset).limit(10)
+            files = await cursor.to_list(length=10)
+            return files, next_offset
+
+        except PyMongoError as e:
+            print(f"MongoDB error: {e}")
+            return [], ''  # Return empty list and '' on error
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            return [], ''
+
+    async def GetLatestAllBy10(self, offset=None):
+        try:
+            total_results = await self.files.count_documents({})
+            offset=int( offset if offset else 0)
+            next_offset = offset + 10
+            if next_offset >= total_results:  # Changed > to >= to handle the last page correctly
+                next_offset = ''
+            cursor = self.files.find({}, {"_id": 1, "poster": 1, "title": 1, "release_date": 1, "description": 1, "genre": 1, "IMDB_id": 1, "type": 1, "file": 1})  # Apply the filter with projection
+
+            cursor.sort('$natural', pymongo.DESCENDING )  # Use pymongo.DESCENDING
+            cursor.skip(offset).limit(10)
+            files = await cursor.to_list(length=10)
+            return files, next_offset
+
+        except PyMongoError as e:
+            print(f"MongoDB error: {e}")
+            return [], ''  # Return empty list and '' on error
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+            return [], ''

Dockerfile

@@ -0,0 +1,33 @@
+FROM python:3.11
+
+# Create and use a non-root user
+RUN useradd -ms /bin/bash admin
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy only the necessary files first for better cache utilization
+COPY requirements.txt /app/requirements.txt
+
+# Install system dependencies and clean up in a single RUN step to reduce layers
+RUN apt-get update -y && apt-get upgrade -y \
+    && apt-get install -y \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy the rest of the application code after installing dependencies for better cache
+COPY . /app
+
+# Set ownership and permissions for the app directory
+RUN chown -R admin:admin /app && chmod -R 777 /app
+
+# Switch to the non-root user for better security
+USER admin
+
+# Expose the port the application will run on
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "-u", "-m", "/app/main.py"]

Tools/__init__.py

@@ -0,0 +1 @@
+from .calculator import size_calculator

Tools/calculator.py

@@ -0,0 +1,10 @@
+def size_calculator(size):
+    if not size:
+        return ""
+    power = 2**10
+    n = 0
+    Dic_powerN = {0: ' ', 1: 'Ki', 2: 'Mi', 3: 'Gi', 4: 'Ti'}
+    while size > power:
+        size /= power
+        n += 1
+    return str(round(size, 2)) + " " + Dic_powerN[n] + 'B'

main.py

@@ -0,0 +1,7 @@
+from fastapi import FastAPI
+
+from server import app
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+motor
+pymongo

server/__init__.py

@@ -0,0 +1,11 @@
+from fastapi import FastAPI
+from .listing import router as listing_router
+
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to the API"}
+
+# Include the router from Listing.py
+app.include_router(listing_router)

server/app.py

@@ -0,0 +1,11 @@
+from fastapi import FastAPI
+from DBFiles.Listing import router as listing_router
+
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to the API"}
+
+# Include the router from Listing.py
+app.include_router(listing_router)

server/config.py

@@ -0,0 +1,9 @@
+from os import environ as env
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class Server:
+    MONGO_DB_URI=str(env.get('DATABASE_URL','mongodb+srv://privateone:6GtbFfj0LK69TQDO@teledb.nxjviws.mongodb.net/?retryWrites=true&w=majority&appName=TeleDB'))
+    DB_NAME=str(env.get('DB_NAME', "FileStream"))

server/listing.py

@@ -0,0 +1,73 @@
+from fastapi import APIRouter, Query
+from fastapi.responses import JSONResponse
+from DBFiles.database import PUBLICFilesDB
+from Tools import size_calculator
+from .config import Server
+#from config import 
+
+router = APIRouter()
+db = PUBLICFilesDB(Server.MONGO_DB_URI, Server.DB_NAME)  # Replace with actual URI and database name
+
+@router.get("/movies")
+async def all_movies(offset: int = Query(0)):
+    results = []
+    files, next_offset = await db.GetMoviesAllBy10(offset=offset)
+    for file in files:
+        results.append({
+            "id": f"{file['_id']}",
+            "poster": file['poster'],
+            "title": file['title'],
+            "release_date": file['release_date'],
+            "genre": file['genre'],
+            "imdb_id": file['IMDB_id'],
+            "type": file['type'],
+            "file_name": file['file']['file_name'],
+            "document_file_id": file['file']['file_id'],
+            "caption": file['file']['file_name'] or "",
+            "size":size_calculator(file['file']['file_size']),
+            "video_type": file['file']['mime_type']        
+        })
+    return JSONResponse(content={"count": len(results), "results": results, "next_offset": next_offset})
+
+@router.get("/series")
+async def all_series(offset: int = Query(0)):
+    results = []
+    files, next_offset = await db.GetSeriesAllBy10(offset=offset)
+    for file in files:
+        results.append({
+            "id": f"{file['_id']}",
+            "poster": file['poster'],
+            "title": file['title'],
+            "release_date": file['release_date'],
+            "genre": file['genre'],
+            "imdb_id": file['IMDB_id'],
+            "type": file['type'],
+            "file_name": file['file']['file_name'],
+            "document_file_id": file['file']['file_id'],
+            "caption": file['file']['file_name'] or "",
+            "size":size_calculator(file['file']['file_size']),
+            "video_type": file['file']['mime_type']        
+        })
+    return JSONResponse(content={"count": len(results), "results": results, "next_offset": next_offset})
+
+@router.get("/latest")
+async def latest(offset: int = Query(0)):
+    results = []
+    files, next_offset = await db.GetLatestAllBy10(offset=offset)
+    for file in files:
+        results.append({
+            "id": f"{file['_id']}",
+            "poster": file['poster'],
+            "title": file['title'],
+            "release_date": file['release_date'],
+            "description": file['description'],
+            "genre": file['genre'],
+            "imdb_id": file['IMDB_id'],
+            "type": file['type'],
+            "file_name": file['file']['file_name'],
+            "document_file_id": file['file']['file_id'],
+            "caption": file['file']['file_name'] or "",
+            "size":size_calculator(file['file']['file_size']),
+            "video_type": file['file']['mime_type']
+        })
+    return JSONResponse(content={"count": len(results), "results": results, "next_offset": next_offset})