Merge pull request 'sentence 추가' (#2) from kdh8219/misskey-random-sentence-bot:feat/sentence-directory into main

Reviewed-on: https://codeberg.org/sunwoo1524/misskey-random-sentence-bot/pulls/2
일단 머지

.example.env

@@ -1,4 +1,5 @@
 MISSKEY_INSTANCE_ADDRESS="worldc.one"
 MISSKEY_ACCESS_TOKEN="BWbGlK6lWuUgXQpFM7igtmbZ30g6BOIc" # it is just a sample
 
-SENTENCES_FILE="./data/sentences.txt" # Putting an absolute path is recommended
+SENTENCES_FILES=['./data/sentences/default.txt'] # Putting an absolute path is recommended
+# SENTENCES_DIRECTORIES=['./data/sentences/'] 

data/sentences/default.txt

@@ -1,3 +1,4 @@
+발 닦고 잠이나 쭉 자고 싶어요
 집에 가고 싶나요? 저도요...
 월드콘은 맛있습니다.
 오늘도 힘내세요!
@@ -11,6 +12,7 @@
 좋은 하루 입니다!
 좋은 하루 보내고 계신가요?
 ㅁㄴㅇㄹ
+asdf
 무슨 말을 할지 생각 중이에요.
 인간 시대의 끝이 도래했다.
 키보드 바꾸고 싶어요.

data/sentences/hylasuwon.txt

@@ -0,0 +1,2 @@
+수원청개구리는 멸종 위기에요!
+수원청개구리, 맹꽁이\n다 멸종위기동물입니다

data/sentences/worldcone_custom_emoji.txt

@@ -0,0 +1,2 @@
+:dlcool:
+:ablobcatdundundun:

docker-compose.example.yml

@@ -8,6 +8,7 @@ services:
     environment:
       - MISSKEY_INSTANCE_ADDRESS=worldc.one
       - MISSKEY_ACCESS_TOKEN=BWbGlK6lWuUgXQpFM7igtmbZ30g6BOIc
-      - SENTENCES_FILE=/data/sentences.txt
+      # - SENTENCES_FILES=['/data/sentences/default.txt']
+      - SENTENCES_DIRECTORIES=['/data/sentences/']
     volumes:
       - ./data:/data

src/choose_sentence.py

@@ -1,14 +1,33 @@
 import random
 import os
 import sys
+import json
+from glob import glob
 
 from dotenv import load_dotenv
 
 load_dotenv() 
 
+_sentences=[]
 try:
-    with open(os.environ["SENTENCES_FILE"], "r") as f:
+    sentence_files=[]
-        _sentences = f.readlines()
+
+    if os.environ.get("SENTENCES_FILE") and os.path.isfile(os.environ["SENTENCES_FILE"]):
+        sentence_files.append(os.environ["SENTENCE_FILES"])
+
+    if os.environ.get("SENTENCES_FILES"):
+        sentence_files.extend(list(filter(os.path.isfile, json.loads(os.environ["SENTENCES_FILES"]))))
+
+    if os.environ.get("SENTENCES_DIRECTORY") and os.path.isdir(os.environ["SENTENCES_DIRECTORY"]):
+        sentence_files.extend(glob(f'{os.environ["SENTENCES_DIRECTORY"]}*.txt'))
+
+    if os.environ.get("SENTENCES_DIRECTORIES"):
+        # 원라인 똥 뿌직(터져도 책임 안짐)
+        sentence_files.extend([glob(f"{_dir}*.txt") for _dir in filter(os.path.isdir, json.loads(os.environ["SENTENCES_DIRECTORIES"]))][0])
+
+    for sentence_file in sentence_files:
+        with open(sentence_file, "r") as f:
+            _sentences.extend(f.readlines())
 except:
     e = sys.exc_info()[1]
     _sentences = [f"에러발생(진짜임): {e}"]