179 lines
5.3 KiB
Python
179 lines
5.3 KiB
Python
import glob, sys
|
|
import random
|
|
import re
|
|
import os
|
|
|
|
print(f"init __name__ {__name__}")
|
|
print(f"init __file__ {os.path.splitext(os.path.basename(__file__))[0]}")
|
|
|
|
#import wildcards
|
|
if __name__ == os.path.splitext(os.path.basename(__file__))[0] :
|
|
from Colors import cprint, Colors as Colors
|
|
else:
|
|
from .Colors import cprint, Colors as Colors
|
|
py_name=os.path.basename(__file__)
|
|
cprint(py_name, Colors.BLUE)
|
|
#print(os.getcwd())
|
|
|
|
# ============================================================
|
|
class wildcards:
|
|
|
|
# 가져올 파일 목록
|
|
card_path=os.path.dirname(__file__)+"\\..\\..\\wildcards\\**\\*.txt"
|
|
#card_path=f"{os.getcwd()}\\wildcards\\**\\*.txt"
|
|
print(f"wildcards card_path {card_path}")
|
|
|
|
# 정규식
|
|
#resub = re.compile(r"(\{)([^\{\}]*)(\})")
|
|
resub = re.compile(r"(\{)(((\d+)|(\d+)?-(\d+)?)?\$\$)?([^\{\}]*)(\})")
|
|
recard = re.compile(r"(__)(.*?)(__)")
|
|
|
|
# 카드 목록
|
|
cards = {}
|
|
seperator=", "
|
|
loop_max=50
|
|
|
|
# | 로 입력된것중 하나 가져오기
|
|
def sub(match):
|
|
#print(f"sub : {(match.groups())}")
|
|
try:
|
|
#m=match.group(2)
|
|
s=match.group(3)
|
|
m=match.group(7).split("|")
|
|
|
|
#print(f"m : {m}")
|
|
if s is None:
|
|
return random.choice(m)
|
|
c=len(m)
|
|
n=int(match.group(4)) if match.group(4) else None
|
|
if n:
|
|
|
|
r=wildcards.seperator.join(random.sample(m,min(n,c)))
|
|
#print(f"n : {n} ; {r}")
|
|
return r
|
|
|
|
n1=match.group(5)
|
|
n2=match.group(6)
|
|
|
|
if n1 or n2:
|
|
a=min(int(n1 if n1 else c), int(n2 if n2 else c),c)
|
|
b=min(max(int(n1 if n1 else 0), int(n2 if n2 else 0)),c)
|
|
#print(f"ab : {a} ; {b}")
|
|
r=wildcards.seperator.join(
|
|
random.sample(
|
|
m,
|
|
random.randint(
|
|
a,b
|
|
)
|
|
)
|
|
)
|
|
#n1=int(match.group(5)) if not match.group(5) is None
|
|
#n2=int(match.group(6)) if not match.group(6) is None
|
|
else:
|
|
r=wildcards.seperator.join(
|
|
random.sample(
|
|
m,
|
|
random.randint(
|
|
0,c
|
|
)
|
|
)
|
|
)
|
|
#print(f"12 : {r}")
|
|
return r
|
|
|
|
|
|
except Exception as e:
|
|
cprint(f"Error : {match.groups()}" ,Colors.YELLOW)
|
|
cprint(f"Exception : {e}" ,Colors.YELLOW)
|
|
return ""
|
|
|
|
|
|
|
|
# | 로 입력된것중 하나 가져오기 반복
|
|
def sub_loop(text):
|
|
bak=text
|
|
for i in range(1, wildcards.loop_max):
|
|
tmp=wildcards.resub.sub(wildcards.sub, bak)
|
|
#print(f"tmp : {tmp}")
|
|
if bak==tmp :
|
|
return tmp
|
|
bak=tmp
|
|
return bak
|
|
|
|
# 카드 중에서 가져오기
|
|
def card(match):
|
|
#print(f"card i : {match.group(2)}")
|
|
if match.group(2) in wildcards.cards :
|
|
r=random.choice(wildcards.cards[match.group(2)])
|
|
else :
|
|
r= match.group(2)
|
|
#print(f"card r : {r}")
|
|
return r
|
|
|
|
|
|
# 카드 중에서 가져오기 반복. | 의 것도 처리
|
|
def card_loop(text):
|
|
bak=text
|
|
for i in range(1, wildcards.loop_max):
|
|
tmp=wildcards.recard.sub(wildcards.card, bak)
|
|
#print(f"card l : {bak}")
|
|
if bak==tmp :
|
|
tmp=wildcards.sub_loop(tmp)
|
|
|
|
if bak==tmp :
|
|
#print(f"card le : {bak}")
|
|
return tmp
|
|
bak=tmp
|
|
#print(f"card le : {bak}")
|
|
return bak
|
|
|
|
# 카드 파일 읽기
|
|
def card_load():
|
|
#cards=wildcards.cards
|
|
card_path=wildcards.card_path
|
|
cards = {}
|
|
#print(f"path : {path}")
|
|
files=glob.glob(card_path, recursive=True)
|
|
#print(f"files : {files}")
|
|
|
|
for file in files:
|
|
basename = os.path.basename(file)
|
|
file_name = os.path.splitext(basename)[0]
|
|
if not file_name in cards:
|
|
cards[file_name]=[]
|
|
#print(f"file_name : {file_name}")
|
|
f = open(file, 'r', encoding='UTF8')
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
line=line.strip()
|
|
# 주석 빈줄 제외
|
|
if line.startswith("#") or len(line)==0:
|
|
continue
|
|
cards[file_name]+=[line]
|
|
#print(f"line : {line}")
|
|
wildcards.cards=cards
|
|
print(f"cards : {len(wildcards.cards)}")
|
|
#print(f"cards : {cards.keys()}")
|
|
|
|
# 실행기
|
|
def run(text):
|
|
|
|
wildcards.card_load()
|
|
|
|
#print(f"text : {text}")
|
|
result=wildcards.card_loop(text)
|
|
#print(f"result : {result}")
|
|
return result
|
|
|
|
# ============================================================
|
|
|
|
#m = p.sub(sub, test)
|
|
#print(m)
|
|
#print(__name__)
|
|
#if __name__ == '__main__' :
|
|
# 테스트용
|
|
test="{3$$a1|{b2|c3|}|d4|{-$$|f|g}|{-2$$h||i}|{1-$$j|k|}}/{$$l|m|}/{0$$n|}"
|
|
print(wildcards.run(test))
|
|
print(wildcards.run("{9$$a|b}"))
|
|
print(wildcards.run("__my__"))
|