Spaces:

Azeez98
/

onthego

Sleeping

App Files Files Community

Azeez98 commited on Jul 22, 2024

Commit

3b509f1

verified ·

1 Parent(s): f67d201

Update app.py

Browse files

Files changed (1) hide show

app.py +277 -277

app.py CHANGED Viewed

@@ -1,277 +1,277 @@
-import os
-import subprocess
-import tarfile
-import asyncio
-from fastapi import FastAPI, HTTPException, Form, UploadFile, File, Depends, Query, Response
-from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse
-from pydantic import BaseModel
-from tempfile import NamedTemporaryFile
-from typing import List
-from pydantic import BaseModel
-app = FastAPI()
-REQUIREMENTS_FILE = "requirements1.txt"
-class DockerImageParams(BaseModel):
-    image_name: str
-    tag: str = 'latest'
-app = FastAPI()
-async def stream_log(file_path):
-    with open(file_path, 'r') as file:
-        while True:
-            line = file.readline()
-            if line:
-                yield line
-            else:
-                await asyncio.sleep(0.1)
-@app.post("/download-dependencies")
-async def download_dependencies(requirements_file: UploadFile = File(...)):
-    try:
-        with NamedTemporaryFile(delete=False) as tmp:
-            tmp.write(requirements_file.file.read())
-            tmp_path = tmp.name
-        # Ensure the directories exist
-        os.makedirs("/tmp/dependencies", exist_ok=True)
-        log_file_path = "/tmp/dependencies/download.log"
-        with open(log_file_path, "w") as log_file:
-            # Download dependencies
-            result = subprocess.run(
-                ["pip", "download", "-r", tmp_path, "-d", "/tmp/dependencies"],
-                stdout=log_file,
-                stderr=log_file
-            )
-        if result.returncode != 0:
-            raise HTTPException(status_code=500, detail="Error downloading dependencies. See log file for details.")
-        # Create a tar file
-        tar_path = "/tmp/dependencies.tar.gz"
-        with tarfile.open(tar_path, "w:gz") as tar:
-            for root, _, files in os.walk("/tmp/dependencies"):
-                for file in files:
-                    file_path = os.path.join(root, file)
-                    tar.add(file_path, arcname=file)
-            tar.add(log_file_path, arcname="download.log")
-        return StreamingResponse(stream_log(log_file_path), media_type="text/plain")
-    except subprocess.CalledProcessError as e:
-        raise HTTPException(status_code=500, detail=f"Error downloading dependencies: {str(e)}")
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-    finally:
-        os.remove(tmp_path)
-class DockerImageParams(BaseModel):
-    image_name: str
-    tag: str = 'latest'
-def download_docker_image(image_name, tag='latest', destination='/tmp/docker-images'):
-    try:
-        os.makedirs(destination, exist_ok=True)
-        tar_path = os.path.join(destination, f'{image_name.replace("/", "_")}.tar')
-        # Remove the existing tar file if it exists
-        if os.path.exists(tar_path):
-            os.remove(tar_path)
-        # Skopeo command to copy the image as a tar file
-        command = [
-            'skopeo', 'copy',
-            f'docker://{image_name}:{tag}',
-            f'docker-archive:{tar_path}'
-        ]
-        subprocess.run(command, check=True)
-        print(f"Image '{image_name}:{tag}' downloaded successfully to {destination}.")
-        return tar_path
-    except subprocess.CalledProcessError as e:
-        print(f"Error downloading image: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Error downloading Docker image: {str(e)}")
-@app.get("/", response_class=HTMLResponse)
-async def read_root():
-    html_content = """
-    <html>
-    <head>
-        <title>Azeez's Help Desk</title>
-        <link href src"/css/index.css" rel="stylesheet">
-        <script src="js/index.js></script>
-    </head>
-    <body>
-        <h1>Welcome to Azeez's Help Desk :)</h1>
-        <div class="tab">
-        <button class="tablink" onclick="openTab(event, 'Docker')" id="defaultOpen">Docker Image Download</button>
-        <button class="tablink" onclick="openTab(event, 'Pip')">Pip Dependencies Download</button>
-        <button class="tablink" onclick="openTab(event, 'Deb')">Debian Packages Download</button>
-        </div>
-        <div id="Docker" class="tabcontent">
-            <h2>Docker Image Download</h2>
-            <form onsubmit="handleDockerFormSubmit(event)">
-                <label for="image_name">Docker Image Name:</label>
-                <input type="text" id="image_name" name="image_name" required><br><br>
-                <label for="tag">Docker Image Tag:</label>
-                <input type="text" id="tag" name="tag" value="latest" required><br><br>
-                <input type="submit" id="docker-submit-button" value="Download Docker Image">
-            </form>
-            <div id="docker-message" style="display: none; margin-top: 10px;"></div>
-        </div>
-        <div id="Pip" class="tabcontent">
-            <h2>Pip Dependencies Download</h2>
-            <form onsubmit="handlePipFormSubmit(event)">
-                <label for="requirements_file">Requirements File:</label>
-                <input type="file" id="requirements_file" name="requirements_file" accept=".txt" required><br><br>
-                <input type="submit" id="pip-submit-button" value="Download Dependencies">
-            </form>
-            <div id="pip-message" style="display: none; margin-top: 10px;"></div>
-        </div>
-        <div id="Deb" class="tabcontent">
-            <h2>Debian Packages Download</h2>
-            <form onsubmit="handleDebFormSubmit(event)">
-                <label for="deb_packages">Debian Package Names (comma-separated):</label>
-                <input type="text" id="deb_packages" name="deb_packages" required><br><br>
-                <input type="submit" id="deb-submit-button" value="Download Debian Packages">
-            </form>
-            <div id="deb-message" style="display: none; margin-top: 10px;"></div>
-        </div>
-        <div id="progress-container">
-                <div id="progress-bar">0%</div>
-        </div>
-    </body>
-    </html>
-    """
-    return HTMLResponse(content=html_content)
-@app.get("/download-docker-image")
-async def download_docker_image_endpoint(image_name: str = Query(...), tag: str = Query('latest')):
-    tar_path = download_docker_image(image_name, tag)
-    file_size = os.path.getsize(tar_path)
-    def iterfile():
-        with open(tar_path, 'rb') as file:
-            while chunk := file.read(1024 * 1024):  # Read in 1 MB chunks
-                yield chunk
-    headers = {
-        "Content-Disposition": f'attachment; filename="{image_name.replace("/", "_")}.tar"',
-        "Content-Length": str(file_size)
-    }
-    return StreamingResponse(iterfile(), media_type='application/x-tar', headers=headers)
-def create_tar_file(files_to_package: List[str], tar_filename: str, destination_dir: str):
-    """
-    Create a tar file containing specified files.
-    Args:
-    - files_to_package (list): List of paths to files to include in the tar file.
-    - tar_filename (str): Name of the tar file to create.
-    - destination_dir (str): Directory to save the tar file.
-    """
-    try:
-        tar_path = os.path.join(destination_dir, tar_filename)
-        with tarfile.open(tar_path, "w:gz") as tar:
-            for file_path in files_to_package:
-                tar.add(file_path, arcname=os.path.basename(file_path))
-        print(f"Created tar file '{tar_filename}' successfully in '{destination_dir}'.")
-        return tar_path
-    except Exception as e:
-        print(f"Error creating tar file: {e}")
-        raise
-def download_deb_packages(package_names: List[str], destination_dir: str) -> List[str]:
-    """
-    Download Debian packages (`.deb`) and their dependencies using `apt-get download`.
-    Args:
-    - package_names (list): List of package names to download.
-    - destination_dir (str): Directory to save downloaded packages.
-    Returns:
-    - List of paths to downloaded `.deb` packages.
-    """
-    try:
-        # Create the destination directory if it doesn't exist
-        os.makedirs(destination_dir, exist_ok=True)
-        downloaded_packages = []
-        # Download each package and its dependencies
-        for package_name in package_names:
-            # Run apt-get update to refresh package index
-            # subprocess.run(['apt-get', 'update'], check=True)
-            # Download the package to the destination directory
-            subprocess.run(['apt-get', 'download', package_name, '-d', destination_dir], check=True)
-            # Build the full path to the downloaded package
-            deb_filename = f"{package_name}.deb"
-            downloaded_packages.append(os.path.join(destination_dir, deb_filename))
-        return downloaded_packages
-    except subprocess.CalledProcessError as e:
-        print(f"Error downloading packages: {e}")
-        raise
-import subprocess
-def download_make_package(destination_dir):
-    try:
-        # Ensure the destination directory exists
-        subprocess.run(['mkdir', '-p', destination_dir])
-        # Download the make package
-        result = subprocess.run(['apt-get', 'download', 'make', '-d', destination_dir], check=True)
-        if result.returncode == 0:
-            print(f"Downloaded make package and dependencies to {destination_dir} successfully.")
-        else:
-            print("Failed to download make package.")
-    except subprocess.CalledProcessError as e:
-        print(f"Error downloading packages: {e}")
-@app.post("/download-deb-packages")
-async def download_deb_packages_handler(deb_packages: str = Form(...)):
-    try:
-        destination_dir = '/tmp/downloaded_packages'
-        subprocess.run(['mkdir', '-p', destination_dir])  # Ensure destination directory exists
-        package_name='make'
-        # Download the package using apt-get
-        result = subprocess.run(['apt-get', 'install', package_name], check=True)
-        # If download is successful, return the downloaded package
-        if result.returncode == 0:
-            tar_filename = f'{package_name}.tar.gz'
-            tar_path = f'{destination_dir}/{tar_filename}'
-            return FileResponse(tar_path, filename=tar_filename)
-        # If download fails, raise HTTPException
-        else:
-            raise HTTPException(status_code=500, detail=f"Failed to download {package_name} package.")
-    except subprocess.CalledProcessError as e:
-        error_message = str(e.stderr)
-        print(f"Error downloading packages: {error_message}")
-        raise HTTPException(status_code=500, detail=f"Error downloading {package_name} package: {error_message}")
-    except Exception as e:
-        print(f"Error: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Failed to download {package_name} package: {str(e)}")
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=5000)

+import os
+import subprocess
+import tarfile
+import asyncio
+from fastapi import FastAPI, HTTPException, Form, UploadFile, File, Depends, Query, Response
+from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse
+from pydantic import BaseModel
+from tempfile import NamedTemporaryFile
+from typing import List
+from pydantic import BaseModel
+app = FastAPI()
+REQUIREMENTS_FILE = "requirements1.txt"
+class DockerImageParams(BaseModel):
+    image_name: str
+    tag: str = 'latest'
+app = FastAPI()
+async def stream_log(file_path):
+    with open(file_path, 'r') as file:
+        while True:
+            line = file.readline()
+            if line:
+                yield line
+            else:
+                await asyncio.sleep(0.1)
+@app.post("/download-dependencies")
+async def download_dependencies(requirements_file: UploadFile = File(...)):
+    try:
+        with NamedTemporaryFile(delete=False) as tmp:
+            tmp.write(requirements_file.file.read())
+            tmp_path = tmp.name
+        # Ensure the directories exist
+        os.makedirs("/tmp/dependencies", exist_ok=True)
+        log_file_path = "/tmp/dependencies/download.log"
+        with open(log_file_path, "w") as log_file:
+            # Download dependencies
+            result = subprocess.run(
+                ["pip", "download", "-r", tmp_path, "-d", "/tmp/dependencies"],
+                stdout=log_file,
+                stderr=log_file
+            )
+        if result.returncode != 0:
+            raise HTTPException(status_code=500, detail="Error downloading dependencies. See log file for details.")
+        # Create a tar file
+        tar_path = "/tmp/dependencies.tar.gz"
+        with tarfile.open(tar_path, "w:gz") as tar:
+            for root, _, files in os.walk("/tmp/dependencies"):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    tar.add(file_path, arcname=file)
+            tar.add(log_file_path, arcname="download.log")
+        return StreamingResponse(stream_log(log_file_path), media_type="text/plain")
+    except subprocess.CalledProcessError as e:
+        raise HTTPException(status_code=500, detail=f"Error downloading dependencies: {str(e)}")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        os.remove(tmp_path)
+class DockerImageParams(BaseModel):
+    image_name: str
+    tag: str = 'latest'
+def download_docker_image(image_name, tag='latest', destination='/tmp/docker-images'):
+    try:
+        os.makedirs(destination, exist_ok=True)
+        tar_path = os.path.join(destination, f'{image_name.replace("/", "_")}.tar')
+        # Remove the existing tar file if it exists
+        if os.path.exists(tar_path):
+            os.remove(tar_path)
+        # Skopeo command to copy the image as a tar file
+        command = [
+            'skopeo', 'copy',
+            f'docker://{image_name}:{tag}',
+            f'docker-archive:{tar_path}'
+        ]
+        subprocess.run(command, check=True)
+        print(f"Image '{image_name}:{tag}' downloaded successfully to {destination}.")
+        return tar_path
+    except subprocess.CalledProcessError as e:
+        print(f"Error downloading image: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error downloading Docker image: {str(e)}")
+@app.get("/", response_class=HTMLResponse)
+async def read_root():
+    html_content = """
+    <html>
+    <head>
+        <title>Azeez's Help Desk</title>
+        <link href src"./css/index.css" rel="stylesheet">
+        <script src="./js/index.js></script>
+    </head>
+    <body>
+        <h1>Welcome to Azeez's Help Desk :)</h1>
+        <div class="tab">
+        <button class="tablink" onclick="openTab(event, 'Docker')" id="defaultOpen">Docker Image Download</button>
+        <button class="tablink" onclick="openTab(event, 'Pip')">Pip Dependencies Download</button>
+        <button class="tablink" onclick="openTab(event, 'Deb')">Debian Packages Download</button>
+        </div>
+        <div id="Docker" class="tabcontent">
+            <h2>Docker Image Download</h2>
+            <form onsubmit="handleDockerFormSubmit(event)">
+                <label for="image_name">Docker Image Name:</label>
+                <input type="text" id="image_name" name="image_name" required><br><br>
+                <label for="tag">Docker Image Tag:</label>
+                <input type="text" id="tag" name="tag" value="latest" required><br><br>
+                <input type="submit" id="docker-submit-button" value="Download Docker Image">
+            </form>
+            <div id="docker-message" style="display: none; margin-top: 10px;"></div>
+        </div>
+        <div id="Pip" class="tabcontent">
+            <h2>Pip Dependencies Download</h2>
+            <form onsubmit="handlePipFormSubmit(event)">
+                <label for="requirements_file">Requirements File:</label>
+                <input type="file" id="requirements_file" name="requirements_file" accept=".txt" required><br><br>
+                <input type="submit" id="pip-submit-button" value="Download Dependencies">
+            </form>
+            <div id="pip-message" style="display: none; margin-top: 10px;"></div>
+        </div>
+        <div id="Deb" class="tabcontent">
+            <h2>Debian Packages Download</h2>
+            <form onsubmit="handleDebFormSubmit(event)">
+                <label for="deb_packages">Debian Package Names (comma-separated):</label>
+                <input type="text" id="deb_packages" name="deb_packages" required><br><br>
+                <input type="submit" id="deb-submit-button" value="Download Debian Packages">
+            </form>
+            <div id="deb-message" style="display: none; margin-top: 10px;"></div>
+        </div>
+        <div id="progress-container">
+                <div id="progress-bar">0%</div>
+        </div>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)
+@app.get("/download-docker-image")
+async def download_docker_image_endpoint(image_name: str = Query(...), tag: str = Query('latest')):
+    tar_path = download_docker_image(image_name, tag)
+    file_size = os.path.getsize(tar_path)
+    def iterfile():
+        with open(tar_path, 'rb') as file:
+            while chunk := file.read(1024 * 1024):  # Read in 1 MB chunks
+                yield chunk
+    headers = {
+        "Content-Disposition": f'attachment; filename="{image_name.replace("/", "_")}.tar"',
+        "Content-Length": str(file_size)
+    }
+    return StreamingResponse(iterfile(), media_type='application/x-tar', headers=headers)
+def create_tar_file(files_to_package: List[str], tar_filename: str, destination_dir: str):
+    """
+    Create a tar file containing specified files.
+    Args:
+    - files_to_package (list): List of paths to files to include in the tar file.
+    - tar_filename (str): Name of the tar file to create.
+    - destination_dir (str): Directory to save the tar file.
+    """
+    try:
+        tar_path = os.path.join(destination_dir, tar_filename)
+        with tarfile.open(tar_path, "w:gz") as tar:
+            for file_path in files_to_package:
+                tar.add(file_path, arcname=os.path.basename(file_path))
+        print(f"Created tar file '{tar_filename}' successfully in '{destination_dir}'.")
+        return tar_path
+    except Exception as e:
+        print(f"Error creating tar file: {e}")
+        raise
+def download_deb_packages(package_names: List[str], destination_dir: str) -> List[str]:
+    """
+    Download Debian packages (`.deb`) and their dependencies using `apt-get download`.
+    Args:
+    - package_names (list): List of package names to download.
+    - destination_dir (str): Directory to save downloaded packages.
+    Returns:
+    - List of paths to downloaded `.deb` packages.
+    """
+    try:
+        # Create the destination directory if it doesn't exist
+        os.makedirs(destination_dir, exist_ok=True)
+        downloaded_packages = []
+        # Download each package and its dependencies
+        for package_name in package_names:
+            # Run apt-get update to refresh package index
+            # subprocess.run(['apt-get', 'update'], check=True)
+            # Download the package to the destination directory
+            subprocess.run(['apt-get', 'download', package_name, '-d', destination_dir], check=True)
+            # Build the full path to the downloaded package
+            deb_filename = f"{package_name}.deb"
+            downloaded_packages.append(os.path.join(destination_dir, deb_filename))
+        return downloaded_packages
+    except subprocess.CalledProcessError as e:
+        print(f"Error downloading packages: {e}")
+        raise
+import subprocess
+def download_make_package(destination_dir):
+    try:
+        # Ensure the destination directory exists
+        subprocess.run(['mkdir', '-p', destination_dir])
+        # Download the make package
+        result = subprocess.run(['apt-get', 'download', 'make', '-d', destination_dir], check=True)
+        if result.returncode == 0:
+            print(f"Downloaded make package and dependencies to {destination_dir} successfully.")
+        else:
+            print("Failed to download make package.")
+    except subprocess.CalledProcessError as e:
+        print(f"Error downloading packages: {e}")
+@app.post("/download-deb-packages")
+async def download_deb_packages_handler(deb_packages: str = Form(...)):
+    try:
+        destination_dir = '/tmp/downloaded_packages'
+        subprocess.run(['mkdir', '-p', destination_dir])  # Ensure destination directory exists
+        package_name='make'
+        # Download the package using apt-get
+        result = subprocess.run(['apt-get', 'install', package_name], check=True)
+        # If download is successful, return the downloaded package
+        if result.returncode == 0:
+            tar_filename = f'{package_name}.tar.gz'
+            tar_path = f'{destination_dir}/{tar_filename}'
+            return FileResponse(tar_path, filename=tar_filename)
+        # If download fails, raise HTTPException
+        else:
+            raise HTTPException(status_code=500, detail=f"Failed to download {package_name} package.")
+    except subprocess.CalledProcessError as e:
+        error_message = str(e.stderr)
+        print(f"Error downloading packages: {error_message}")
+        raise HTTPException(status_code=500, detail=f"Error downloading {package_name} package: {error_message}")
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to download {package_name} package: {str(e)}")
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=5000)