Update main.py

main.py

@@ -1,94 +1,3 @@
-# import enkacard
-# import enkacard.encbanner
-# import enkanetwork
-# import asyncio
-# from flask import Flask, jsonify, request
-# import requests
-# from io import BytesIO
-# from enkanetwork import EnkaNetworkAPI
-# import os
-
-# enka_update = EnkaNetworkAPI()
-
-# async def update_genshin():
-#     async with enka_update:
-#         await enka_update.update_assets(lang = ["EN"], path="/tmp")
-
-# app = Flask(__name__)
-
-# async def card(id, designtype):
-#     async with enkacard.encbanner.ENC(uid = str(id)) as encard:
-#         return await encard.creat(template = (2 if str(designtype) == "2" else 1))
-    
-# @app.route("/<int:id>")
-# def characters(id):
-#     try:
-#         characters = []
-#         result = asyncio.run(card(id, request.args.get('design')))
-#         for dt in result.card:
-#          with BytesIO() as byte_io: 
-#             print("prcess" + dt.name)
-#             dt.card.save(byte_io, "PNG")
-#             byte_io.seek(0)
-#             image_url = upload_image(byte_io)
-            
-#             characters.append({
-#                 "name": dt.name,
-#                 "url": image_url
-#             })
-#             print("done" + dt.name)
-#         return jsonify({'response': characters}) 
-#     except enkanetwork.exception.VaildateUIDError:
-#         # return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400 
-
-#         return jsonify({'error': 'Invalid UID. Please check your UID.'})
-
-#     except enkacard.enc_error.ENCardError:
-#         # return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400 
-#         return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'})
-#     except Exception as e:
-
-#         return jsonify({'error': 'UNKNOWN ERR' + e.__str__()})
-
-
-# @app.route("/")
-# def hello_world():
-#     return 'AMERICA YA HALLO!!'
-
-# @app.route("/update_char")
-# def upload():
-#     data_dir = "/tmp/data"
-
-#     if not os.path.exists(data_dir):
-#         os.makedirs(data_dir)
-
-#     data_dir = "/tmp/langs"
-
-#     if not os.path.exists(data_dir):
-#         os.makedirs(data_dir)
-#     asyncio.run(update_genshin())
-#     return 'Update smth ig!!'
-
-# def upload_image(data):
-#     url = "https://telegra.ph/upload"
-#     files = {'file': ('file', data, "image/png")}  
-#     response = requests.post(url, files=files)
-    
-#     if response.status_code != 200:
-#         raise Exception(f"HTTP Error: {response.status_code}")
-    
-#     try:
-#         body = response.json()
-#         if isinstance(body, list) and 'src' in body[0]:
-#             return "https://telegra.ph" + body[0]['src']
-#         else:
-#             raise Exception(f"Telegraph error: {body.get('error', 'Unknown error')}")
-#     except (ValueError, KeyError, IndexError) as e:
-#         raise Exception(f"Failed to parse response: {str(e)}")
-    
-# if __name__ == "__main__":
-#     app.run(debug=False, use_reloader=False)
-
 import enkacard
 import enkacard.encbanner
 import enkanetwork
@@ -98,44 +7,43 @@ import requests
 from io import BytesIO
 from enkanetwork import EnkaNetworkAPI
 import os
-import base64
 
 enka_update = EnkaNetworkAPI()
 
-async def update_genshin():
-    async with enka_update:
-        await enka_update.update_assets(lang=["EN"], path="/tmp")
-
 app = Flask(__name__)
 
 async def card(id, designtype):
-    async with enkacard.encbanner.ENC(uid=str(id)) as encard:
-        return await encard.creat(template=(2 if str(designtype) == "2" else 1))
-
+    async with enkacard.encbanner.ENC(uid = str(id)) as encard:
+        return await encard.creat(template = (2 if str(designtype) == "2" else 1))
+    
 @app.route("/<int:id>")
 def characters(id):
     try:
         characters = []
         result = asyncio.run(card(id, request.args.get('design')))
         for dt in result.card:
-            with BytesIO() as byte_io:
-                print("processing " + dt.name)
-                dt.card.save(byte_io, "PNG")
-                byte_io.seek(0)
-                image_url = upload_image(byte_io)
-                
-                characters.append({
-                    "name": dt.name,
-                    "url": image_url
-                })
-        return jsonify({'response': characters})
+         with BytesIO() as byte_io: 
+            dt.card.save(byte_io, "PNG")
+            byte_io.seek(0)
+            image_url = upload_image(byte_io)
+            
+            characters.append({
+                "name": dt.name,
+                "url": image_url
+            })
+        return jsonify({'response': characters}) 
     except enkanetwork.exception.VaildateUIDError:
-        return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400
+        # return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400 
+
+        return jsonify({'error': 'Invalid UID. Please check your UID.'})
+
     except enkacard.enc_error.ENCardError:
-        return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400
+        # return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400 
+        return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'})
     except Exception as e:
-        print(e)
-        return jsonify({'error': 'UNKNOWN ERR: ' + str(e)}), 500
+
+        return jsonify({'error': 'UNKNOWN ERR' + e.__str__()})
+
 
 @app.route("/")
 def hello_world():
@@ -143,46 +51,34 @@ def hello_world():
 
 @app.route("/update_char")
 def upload():
-    data_dirs = ["/tmp/data", "/tmp/langs"]
+    data_dir = "/tmp/data"
+
+    if not os.path.exists(data_dir):
+        os.makedirs(data_dir)
 
-    for directory in data_dirs:
-        if not os.path.exists(directory):
-            os.makedirs(directory)
+    data_dir = "/tmp/langs"
 
+    if not os.path.exists(data_dir):
+        os.makedirs(data_dir)
     asyncio.run(update_genshin())
-    return 'Update completed!'
+    return 'Update smth ig!!'
 
 def upload_image(data):
-    """
-    Convert image data to a base64-encoded string.
+    url = "https://telegra.ph/upload"
+    files = {'file': ('file', data, "image/png")}  
+    response = requests.post(url, files=files)
     
-    Parameters:
-    - data: A BytesIO object containing the image data.
-
-    Returns:
-    - A base64-encoded string of the image.
-    """
-    # Convert the image data to a base64-encoded string
-    image_base64 = base64.b64encode(data.getvalue()).decode('utf-8')
-    
-    # Return the base64-encoded string
-    return image_base64
-# def upload_image(data):
-#     url = "https://telegra.ph/upload"
-#     files = {'file': ('file', data, "image/png")}
-#     response = requests.post(url, files=files)
+    if response.status_code != 200:
+        raise Exception(f"HTTP Error: {response.status_code}")
     
-#     if response.status_code != 200:
-#         raise Exception(f"HTTP Error: {response.status_code}")
+    try:
+        body = response.json()
+        if isinstance(body, list) and 'src' in body[0]:
+            return "https://telegra.ph" + body[0]['src']
+        else:
+            raise Exception(f"Telegraph error: {body.get('error', 'Unknown error')}")
+    except (ValueError, KeyError, IndexError) as e:
+        raise Exception(f"Failed to parse response: {str(e)}")
     
-#     try:
-#         body = response.json()
-#         if isinstance(body, list) and 'src' in body[0]:
-#             return "https://telegra.ph" + body[0]['src']
-#         else:
-#             raise Exception(f"Telegraph error: {body.get('error', 'Unknown error')}")
-#     except (ValueError, KeyError, IndexError) as e:
-#         raise Exception(f"Failed to parse response: {str(e)}")
-
 if __name__ == "__main__":
     app.run(debug=False, use_reloader=False)