Commit 66e4279a774aea5b996c62b321ce94e7e7296fdf

Authored by quillotm
1 parent 83e2994d06
Exists in master

Little error solved

Showing 1 changed file with 3 additions and 1 deletions Inline Diff

1 1
2 import argparse 2 import argparse
3 from utils import SubCommandRunner 3 from utils import SubCommandRunner
4 from core.data import read_lst 4 from core.data import read_lst
5 import os 5 import os
6 6
7 7
8 8
9 def lst2wav(lst: str, root: str, outfile: str): 9 def lst2wav(lst: str, root: str, outfile: str):
10 lst_ = read_lst(lst) 10 lst_ = read_lst(lst)
11 11
12 with open(outfile, "w") as f: 12 with open(outfile, "w") as f:
13 for id_ in lst_: 13 for id_ in lst_:
14 # Split id 14 # Split id
15 splited = id_.split(",") 15 splited = id_.replace("\n", "").split(",")
16 16
17 # Get language 17 # Get language
18 language = None 18 language = None
19 if splited[0] == "en-us": 19 if splited[0] == "en-us":
20 language = "english" 20 language = "english"
21 elif splited[0] == "": 21 elif splited[0] == "":
22 language = "french" 22 language = "french"
23 print(splited[0])
24 print(language)
23 25
24 # Create wav filename 26 # Create wav filename
25 splited3 = splited[3].split(".") 27 splited3 = splited[3].split(".")
26 wav = "-".join(splited3) + ".wav" 28 wav = "-".join(splited3) + ".wav"
27 29
28 # Create full path 30 # Create full path
29 full_path = os.path.join(language, "wave-audio", wav) 31 full_path = os.path.join(language, "wave-audio", wav)
30 32
31 f.write(f"{id_} {full_path}\n") 33 f.write(f"{id_} {full_path}\n")
32 pass 34 pass
33 pass 35 pass
34 36
35 37
36 if __name__ == '__main__': 38 if __name__ == '__main__':
37 # Main parser 39 # Main parser
38 parser = argparse.ArgumentParser(description="Skyrim data-management commands") 40 parser = argparse.ArgumentParser(description="Skyrim data-management commands")
39 subparsers = parser.add_subparsers(title="action") 41 subparsers = parser.add_subparsers(title="action")
40 42
41 parser_lst2wav = subparsers.add_parser("lst2wav", help="Generate wav file from lst") 43 parser_lst2wav = subparsers.add_parser("lst2wav", help="Generate wav file from lst")
42 parser_lst2wav.add_argument("--lst", type=str, help="list file .lst") 44 parser_lst2wav.add_argument("--lst", type=str, help="list file .lst")
43 parser_lst2wav.add_argument("--root", type=str, help="root directory with audio files") 45 parser_lst2wav.add_argument("--root", type=str, help="root directory with audio files")
44 parser_lst2wav.add_argument("--outfile", type=str, help="output wav.scp-like file") 46 parser_lst2wav.add_argument("--outfile", type=str, help="output wav.scp-like file")
45 parser_lst2wav.set_defaults(which="lst2wav") 47 parser_lst2wav.set_defaults(which="lst2wav")
46 48
47 # Parse 49 # Parse
48 args = parser.parse_args() 50 args = parser.parse_args()
49 51
50 # Run commands 52 # Run commands
51 runner = SubCommandRunner({ 53 runner = SubCommandRunner({
52 "lst2wav" : lst2wav 54 "lst2wav" : lst2wav
53 }) 55 })
54 56
55 runner.run(args.which, args.__dict__, remove="which") 57 runner.run(args.which, args.__dict__, remove="which")
56 58