Initial commit

home-server/printing/server/src/print_api.py

@@ -0,0 +1,69 @@
+import os
+from pprint import pprint
+import tempfile
+from fastapi import FastAPI, File, UploadFile, Request
+import cups
+from fastapi.responses import HTMLResponse
+
+app = FastAPI()
+
+
+# @app.post("/print/")
+# async def print_document(file: UploadFile = File(...)):
+#     temp_file = tempfile.NamedTemporaryFile(delete=False)
+#     temp_file.write(await file.read())
+#     temp_file.close()
+    
+#     conn = cups.Connection(host="server.alexmickelson.guru")
+    
+#     printers = conn.getPrinters()
+#     print(file.filename)
+#     print(temp_file.name)
+#     pprint(printers)
+#     for printer in printers:
+#         print(printer, printers[printer]["device-uri"])
+
+#     default_printer = list(printers.keys())[0]
+
+
+#     job_id = conn.printFile(default_printer, temp_file.name, f"FastAPI Print Job for {temp_file.name}", {})
+#     os.unlink(temp_file.name)
+
+#     return {"job_id": job_id, "file_name": file.filename}
+
+
+@app.post("/print/")
+async def print_document(file: UploadFile = File(...)):
+    # Save the uploaded file to a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False)
+    temp_file.write(await file.read())
+    temp_file.close()
+
+    # Connect to the CUPS server on the host (use default CUPS connection)
+    conn = cups.Connection()  # This will connect to localhost CUPS
+
+    # Get the list of available printers
+    printers = conn.getPrinters()
+    print(file.filename)
+    print(temp_file.name)
+    pprint(printers)
+    for printer in printers:
+        print(printer, printers[printer]["device-uri"])
+
+    # Use the default printer (first one in the list)
+    default_printer = list(printers.keys())[0]
+
+    # Print the file
+    job_id = conn.printFile(default_printer, temp_file.name, f"FastAPI Print Job for {temp_file.name}", {})
+    
+    # Clean up the temporary file
+    os.unlink(temp_file.name)
+
+    return {"job_id": job_id, "file_name": file.filename}
+
+@app.get("/", response_class=HTMLResponse)
+async def read_root(request: Request):
+    with open('src/index.html', 'r') as f:
+        html_content = f.read()
+        return HTMLResponse(content=html_content, status_code=200)
+