feat(choose_sentence): 다중 파일, 다중 디렉토리 성택기능 추가

This commit is contained in:
kdh8219 2023-12-23 19:20:18 +09:00
parent 12f40fc30f
commit 75a1659734
Signed by: kdh8219
GPG key ID: 9B901BE907D1862E
3 changed files with 14 additions and 4 deletions

View file

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

View file

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

View file

@ -1,6 +1,7 @@
import random import random
import os import os
import sys import sys
import json
from glob import glob from glob import glob
from dotenv import load_dotenv from dotenv import load_dotenv
@ -10,12 +11,20 @@ load_dotenv()
_sentences=[] _sentences=[]
try: try:
sentence_files=[] sentence_files=[]
if os.environ.get("SENTENCES_FILE") and os.path.isfile(os.environ["SENTENCES_FILE"]): if os.environ.get("SENTENCES_FILE") and os.path.isfile(os.environ["SENTENCES_FILE"]):
sentence_files.append(os.environ["SENTENCE_FILES"]) 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"]): if os.environ.get("SENTENCES_DIRECTORY") and os.path.isdir(os.environ["SENTENCES_DIRECTORY"]):
sentence_files.extend(glob(f'{os.environ["SENTENCES_DIRECTORY"]}*.txt')) 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"]))])
for sentence_file in sentence_files: for sentence_file in sentence_files:
with open(sentence_file, "r") as f: with open(sentence_file, "r") as f:
_sentences.extend(f.readlines()) _sentences.extend(f.readlines())