From 8bfae82fa06f2809e70b0ad9110c2b8222ee2174 Mon Sep 17 00:00:00 2001
From: Tony Rewin <anton.rewin@gmail.com>
Date: Thu, 5 Oct 2023 11:51:27 +0300
Subject: [PATCH] starlette

---
 README.md        | 10 +++++++++-
 main.py          | 33 +++++++++++++++++++++------------
 requirements.txt |  7 ++++---
 3 files changed, 34 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index d66fa3a..2a26ff9 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,12 @@
-#### ENV setup
+## `uploader`: загружает файлы в IPFS
+
+Использует:
+- python3
+- uvicorn ASGI -> WSGI
+- starlette ASGI server
+- boto3 s3 client
+
+#### Переменные среды
 
  - STORJ_ACCESS_KEY
  - STORJ_SECRET_KEY
diff --git a/main.py b/main.py
index 14ca0a7..3317564 100644
--- a/main.py
+++ b/main.py
@@ -3,7 +3,10 @@ import tempfile
 import uuid
 import boto3
 from botocore.exceptions import BotoCoreError, ClientError
-from aiohttp import web
+from starlette.applications import Starlette
+from starlette.responses import JSONResponse
+from starlette.routing import Route
+from starlette.requests import Request
 
 STORJ_ACCESS_KEY = os.environ.get('STORJ_ACCESS_KEY')
 STORJ_SECRET_KEY = os.environ.get('STORJ_SECRET_KEY')
@@ -11,22 +14,25 @@ STORJ_END_POINT = os.environ.get('STORJ_END_POINT')
 STORJ_BUCKET_NAME = os.environ.get('STORJ_BUCKET_NAME')
 CDN_DOMAIN = os.environ.get('CDN_DOMAIN')
 
+async def upload_handler(request: Request):
+    form = await request.form()
+    file = form.get('file')
 
-async def upload_handler(request):
-    reader = await request.multipart()
-    file = await reader.next()
     if file is None:
-        return web.json_response({'error': 'No file uploaded'}, status=400)
+        return JSONResponse({'error': 'No file uploaded'}, status_code=400)
+
     file_name, file_extension = os.path.splitext(file.filename)
     key = str(uuid.uuid4()) + file_extension
+
     s3 = boto3.client('s3',
                       aws_access_key_id=STORJ_ACCESS_KEY,
                       aws_secret_access_key=STORJ_SECRET_KEY,
                       endpoint_url=STORJ_END_POINT)
+
     try:
         with tempfile.NamedTemporaryFile() as tmp_file:
             while True:
-                chunk = await file.read_chunk()  # 8192 bytes by default.
+                chunk = await file.read(8192)  # 8192 bytes by default.
                 if not chunk:
                     break
                 tmp_file.write(chunk)
@@ -42,15 +48,18 @@ async def upload_handler(request):
 
         url = 'http://' + CDN_DOMAIN + '/' + key
 
-        return web.json_response({'url': url, 'originalFilename': file.filename})
+        return JSONResponse({'url': url, 'originalFilename': file.filename})
 
     except (BotoCoreError, ClientError) as e:
         print(e)
-        return web.json_response({'error': 'Failed to upload file'}, status=500)
+        return JSONResponse({'error': 'Failed to upload file'}, status_code=500)
 
+routes = [
+    Route('/upload', upload_handler, methods=['POST']),
+]
+
+app = Starlette(debug=True, routes=routes)
 
 if __name__ == "__main__":
-    app = web.Application()
-    app.router.add_post('/upload', upload_handler)
-
-    web.run_app(app)
\ No newline at end of file
+    import uvicorn
+    uvicorn.run(app, host='0.0.0.0', port=8000)
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 7c2fa69..d04c8c6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
-aiohttp~=3.8.5
-boto3~=1.28.56
-botocore~=1.31.56
\ No newline at end of file
+uvicorn
+starlette
+boto3
+botocore
\ No newline at end of file