2

main.py

@@ -1,3 +1,94 @@
+# 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
@@ -7,47 +98,44 @@ 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")
+        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: 
-            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:
+                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})
     except enkanetwork.exception.VaildateUIDError:
-        # return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400 
-
-        return jsonify({'error': 'Invalid UID. Please check your UID.'})
-
+        return jsonify({'error': 'Invalid UID. Please check your UID.'}), 400
     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.'})
+        return jsonify({'error': 'Enable display of the showcase in the game or add characters there.'}), 400
     except Exception as e:
-
-        return jsonify({'error': 'UNKNOWN ERR' + e.__str__()})
-
+        print(e)
+        return jsonify({'error': 'UNKNOWN ERR: ' + str(e)}), 500
 
 @app.route("/")
 def hello_world():
@@ -55,34 +143,46 @@ def hello_world():
 
 @app.route("/update_char")
 def upload():
-    data_dir = "/tmp/data"
-
-    if not os.path.exists(data_dir):
-        os.makedirs(data_dir)
+    data_dirs = ["/tmp/data", "/tmp/langs"]
 
-    data_dir = "/tmp/langs"
+    for directory in data_dirs:
+        if not os.path.exists(directory):
+            os.makedirs(directory)
 
-    if not os.path.exists(data_dir):
-        os.makedirs(data_dir)
     asyncio.run(update_genshin())
-    return 'Update smth ig!!'
+    return 'Update completed!'
 
 def upload_image(data):
-    url = "https://telegra.ph/upload"
-    files = {'file': ('file', data, "image/png")}  
-    response = requests.post(url, files=files)
+    """
+    Convert image data to a base64-encoded string.
     
-    if response.status_code != 200:
-        raise Exception(f"HTTP Error: {response.status_code}")
+    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')
     
-    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)}")
+    # 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}")
+    
+#     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)
+    app.run(debug=False, use_reloader=False)