first commit

test.py

@@ -0,0 +1,36 @@
+import os
+from pathlib import Path
+from models.generator import Generator
+from models.emoji_db import check_emoji, update_description
+from controllers.localfile_handles import get_emoji_count, yield_emoji_path
+from controllers.compress_image import compress_image
+from models.logger import setup_logger
+
+logger = setup_logger()
+
+emoji_folders = "D:\\Python\\mcunc\\EmojiTextGenerator\\emojis\\"
+emoji = yield_emoji_path(emoji_folders)
+
+api_key = os.getenv('DASHSCOPE_API_KEY')
+
+def main():
+    gen = Generator(api_key)
+    for f in range(get_emoji_count(emoji_folders)):
+        emoji_file = next(emoji)
+        emoji_uuid = ''
+        p = Path(emoji_file)
+        if p.suffix == ".jpg" or p.suffix == ".jpeg":
+            emoji_uuid = p.stem
+        # print(emoji_uuid)
+        image = compress_image(emoji_file)
+
+        # if not check_emoji(emoji_uuid):
+        #     logger.error(f"图片 {emoji_uuid} 不在数据库中！")
+        # else:
+        description = gen.process_single_image(image)
+        if update_description(emoji_uuid, description)
+            logger.info(f"图片 {emoji_file} 的描述词生成完毕， 其 description 为: {description}")
+
+
+if __name__ == "__main__":
+    main()