bizydraft 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

bizydraft/__init__.py

@@ -1,6 +1 @@
-
-from .hijack_nodes import hijack_nodes
-
-__all__ = [
-    "hijack_nodes",
-]
+# 在这里如果 import 子包，容易引起 prestartup_script.py 的报错

bizydraft/hijack_nodes.py

@@ -8,6 +8,7 @@ except ImportError:
     )
     exit(1)
 
+
 class BizyDraftLoadImage(LoadImage):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -21,7 +22,5 @@ def hijack_nodes():
     if "LoadImage" in NODE_CLASS_MAPPINGS:
         del NODE_CLASS_MAPPINGS["LoadImage"]
     NODE_CLASS_MAPPINGS["LoadImage"] = BizyDraftLoadImage
-    
-    logger.info(
-        "[BizyDraft] Hijacked LoadImage node to BizyDraftLoadImage."
-    )
+
+    logger.info("[BizyDraft] Hijacked LoadImage node to BizyDraftLoadImage.")

bizydraft/hijack_routes.py

@@ -29,9 +29,7 @@ BIZYDRAFT_CHUNK_SIZE = int(os.getenv("BIZYDRAFT_CHUNK_SIZE", 1024 * 16))  # 16KB
 
 
 async def view_image(request, old_handler):
-    logger.debug(
-        f"Received request for /view with query: {request.rel_url.query}"
-    )
+    logger.debug(f"Received request for /view with query: {request.rel_url.query}")
     if "filename" not in request.rel_url.query:
         logger.warning("'filename' not provided in query string, returning 404")
         return web.Response(status=404, text="'filename' not provided in query string")
@@ -154,12 +152,12 @@ def hijack_routes():
 
             routes._items[idx] = web.get("/view", new_handler)
             routes._items[idx].kwargs.clear()
-            logger.info(
-                "Hijacked /view route to handle image, video and 3D streaming"
-            )
+            logger.info("Hijacked /view route to handle image, video and 3D streaming")
             break
     for idx, route in enumerate(routes._items):
         if route.path == "/prompt" and route.method == "POST":
             routes._items[idx] = web.post("/prompt", post_prompt)
-            logger.info("Hijacked /prompt route to handle prompt validation but not execution")
+            logger.info(
+                "Hijacked /prompt route to handle prompt validation but not execution"
+            )
             break

bizydraft/server.py

@@ -18,6 +18,11 @@ _SERVER_MODE_HC_FLAG = True
 
 BIZYAIR_MAGIC_STRING = os.getenv("BIZYAIR_MAGIC_STRING", "QtDtsxAc8JI1bTb7")
 
+if BIZYAIR_MAGIC_STRING == "QtDtsxAc8JI1bTb7":
+    logger.warning(
+        "BIZYAIR_MAGIC_STRING is not set, using default value. This is insecure and should be changed in production!"
+    )
+
 
 class BizyDraftServer:
     def __init__(self):

{bizydraft-0.1.0.dist-info → bizydraft-0.1.2.dist-info}/METADATA

@@ -1,3 +1,4 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.1.0
+Version: 0.1.2
+Summary: bizydraft

{bizydraft-0.1.0.dist-info → bizydraft-0.1.2.dist-info}/RECORD

@@ -1,8 +1,8 @@
-bizydraft/__init__.py,sha256=qV8OhBrI5PgWcmvkYd5E0L2IiYjtbDSK3zyWpKspv0w,74
-bizydraft/hijack_nodes.py,sha256=CrvATjtlfgvlCcd-7TawWCBlVwBVvsnNCFgGgccuLb8,752
-bizydraft/hijack_routes.py,sha256=rXCtb9T0eZ2FCV9DymkIV9xlXGcgNU86lcxFubGRvxw,6152
+bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
+bizydraft/hijack_nodes.py,sha256=YlshHCRu7Bg5jBclaTuEfBXdbNNw-XZ77BciD2SRN5A,735
+bizydraft/hijack_routes.py,sha256=tzSIY-3wLiwoHALf7MswjAF5FqDSmXynGjpfTgRdkxE,6138
 bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
-bizydraft/server.py,sha256=wbpK_fJVgnvYHQieZvFN_TCvrdyN5Rsmv0DL9waoIXU,1166
+bizydraft/server.py,sha256=ES1fBYr_Ufv_YWsynLck1ln9bXTGjK3KXas1CK9WBqM,1358
 bizydraft/static/js/handleStyle.js,sha256=RJbDRZMeb3_32JqPfOzxnZGVONb-WnXYrQDxIeu0Miw,1263
 bizydraft/static/js/hookLoadImage.js,sha256=cEaqedTnngieI-Kb0PVgLor1nVDkFQx1_7AiurwtRwI,3878
 bizydraft/static/js/main.js,sha256=cZ-7wR9T8aNLzIrTjc-g9xVZf7z6TXWl1zhP_wXSSVo,150
@@ -10,7 +10,7 @@ bizydraft/static/js/postEvent.js,sha256=4PVl929jU1m3ooKA1lFXNWgv62mtKAFnhqbsHVBU
 bizydraft/static/js/socket.js,sha256=iNxbUPSQluIlx1x0MIVL7UJ35BXqLH7Z2s8vmfmbyyc,2643
 bizydraft/static/js/tool.js,sha256=IlDVjddodT9DtfdIGiIzSIKdxbUMztDLQ4jI0uFJCwk,206
 bizydraft/static/js/uploadFile.js,sha256=7PXMYRlgxgDWFFbsaasjf3TelsW6r9mD1TFTjQI6X0E,4957
-bizydraft-0.1.0.dist-info/METADATA,sha256=XouCMG6uu1cvKr8HJvUTPpx4EqAtJMlgYfBXmC2d2tE,53
-bizydraft-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.1.0.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.1.0.dist-info/RECORD,,
+bizydraft-0.1.2.dist-info/METADATA,sha256=d7bj6AAM84FLg8Chv_RQgpBcK-uJO77J8jcgqs7aLZs,72
+bizydraft-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.1.2.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.1.2.dist-info/RECORD,,