Repository navigation
Expand file tree
/
Copy pathconfig-generator.py
More file actions
1063 lines (890 loc) · 47.9 KB
/
Copy pathconfig-generator.py
File metadata and controls
1063 lines (890 loc) · 47.9 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/usr/bin/env python3
import os
import re
import json
from pathlib import Path
import ast
def read_example_config():
"""Read the example config file and return its structure and comments"""
example_path = Path("data/example-config.py")
comments = {}
if not example_path.exists():
print("[!] Warning: Could not find data/example-config.py")
print("[i] Using built-in default structure instead")
return None, comments
try:
with open(example_path, "r", encoding="utf-8") as file:
lines = file.readlines()
current_comments = []
key_stack = []
indent_stack = [0]
for idx, line in enumerate(lines):
line = line.rstrip("\n")
stripped = line.lstrip()
indent = len(line) - len(stripped)
# Track nesting for fully qualified keys
if "{" in stripped and ":" in stripped:
key = stripped.split(":", 1)[0].strip().strip('"\'')
while indent_stack and indent <= indent_stack[-1]:
key_stack.pop()
indent_stack.pop()
key_stack.append(key)
indent_stack.append(indent)
elif "}" in stripped:
while indent_stack and indent <= indent_stack[-1]:
if key_stack: # Avoid popping from empty list
key_stack.pop()
indent_stack.pop()
if stripped.startswith("#"):
current_comments.append(stripped)
elif ":" in stripped and not stripped.startswith("{"):
key = stripped.split(":", 1)[0].strip().strip('"\'')
# Build fully qualified key path
fq_key = ".".join(key_stack + [key]) if key_stack else key
if current_comments:
comments[key] = list(current_comments)
comments[fq_key] = list(current_comments)
current_comments = []
elif not stripped: # Empty line
pass # Keep the comments for the next key
elif stripped in ["},", "}"]:
pass # Keep the comments for the next key
else:
current_comments = [] # Clear comments on other lines
# Extract the config dict from the file content
content = ''.join(lines)
match = re.search(r"config\s*=\s*({.*})", content, re.DOTALL)
if not match:
print("[!] Warning: Could not parse example config")
return None, comments
config_dict_str = match.group(1)
example_config = ast.literal_eval(config_dict_str)
print("[✓] Successfully loaded example config template")
return example_config, comments
except Exception as e:
print(f"[!] Error parsing example config: {str(e)}")
return None, comments
def load_existing_config():
"""Load an existing config file if available"""
config_paths = [
Path("data/config.py"),
Path("data/config1.py")
]
for path in config_paths:
if path.exists():
try:
with open(path, "r", encoding="utf-8") as file:
content = file.read()
# Extract the config dict from the file
match = re.search(r"config\s*=\s*({.*})", content, re.DOTALL)
if match:
config_dict_str = match.group(1)
# Convert to proper Python dict
config_dict = ast.literal_eval(config_dict_str)
print(f"\n[✓] Found existing config at {path}")
return config_dict, path
except Exception as e:
print(f"\n[!] Error loading config from {path}: {e}")
return None, None
def validate_config(existing_config, example_config):
"""
Validate the existing config against the example structure.
Returns a cleaned version with only valid keys.
"""
if not existing_config or not example_config:
return existing_config
unexpected_keys = []
# Helper function to find unexpected keys at any level
def find_unexpected_keys(existing_section, example_section, path=""):
if not isinstance(existing_section, dict) or not isinstance(example_section, dict):
return
for key in existing_section:
current_path = f"{path}.{key}" if path else key
if key not in example_section:
unexpected_keys.append((current_path, existing_section, key))
elif isinstance(existing_section[key], dict) and isinstance(example_section.get(key), dict):
# Recursively check nested dictionaries
find_unexpected_keys(existing_section[key], example_section[key], current_path)
# Check main sections first
for section in existing_config:
if section not in example_config:
unexpected_keys.append((section, existing_config, section))
elif isinstance(existing_config[section], dict) and isinstance(example_config[section], dict):
# Check keys within valid sections
find_unexpected_keys(existing_config[section], example_config[section], section)
# If unexpected keys were found, ask about each one individually
if unexpected_keys:
print("\n[!] The following keys in your existing configuration are not in the example config:")
for i, (key_path, parent_dict, key) in enumerate(unexpected_keys):
print(f" {i+1}. {key_path}")
print("\n\n[i] The keys have been removed or renamed.")
print("[i] You can choose what to do with each key:")
for i, (key_path, parent_dict, key) in enumerate(unexpected_keys):
value = parent_dict[key]
value_display = str(value)
if isinstance(value, dict):
value_display = "{...}" # Just show placeholder for dictionaries
# Handle nested structures by limiting display length
if len(value_display) > 50:
value_display = value_display[:47] + "..."
print(f"\nKey {i+1}/{len(unexpected_keys)}: {key_path} = {value_display}")
keep = input("Keep this key? (y/N): ").lower()
# Remove the key if user chooses not to keep it
if keep == "y":
print(f"[i] Keeping key: {key_path}")
else:
print(f"[i] Removing key: {key_path}")
del parent_dict[key]
return existing_config
# Return original if no unexpected keys
return existing_config
def find_missing_keys(existing_config, example_config):
"""Find keys that exist in example config but are missing in existing config"""
missing_keys = []
# Helper function to find missing keys at any level
def find_missing_recursive(example_section, existing_section, path=""):
if not isinstance(example_section, dict) or not isinstance(existing_section, dict):
return
for key in example_section:
current_path = f"{path}.{key}" if path else key
if key not in existing_section:
missing_keys.append(current_path)
elif isinstance(example_section[key], dict) and isinstance(existing_section.get(key), dict):
# Recursively check nested dictionaries
find_missing_recursive(example_section[key], existing_section[key], current_path)
# Check main sections first
for section in example_config:
if section not in existing_config:
missing_keys.append(section)
elif isinstance(example_config[section], dict) and isinstance(existing_config[section], dict):
# Check keys within valid sections
find_missing_recursive(example_config[section], existing_config[section], section)
return missing_keys
def get_user_input(prompt, default="", is_password=False, is_announce_url=False, existing_value=None):
"""Get input from user with default value and optional existing value"""
display = prompt
# If we have an existing value, show it as an option
if existing_value is not None:
# For password fields: show first 6 chars and mask the rest
if is_password and existing_value:
visible_part = existing_value[:6]
masked_part = "*" * min(8, max(0, len(existing_value) - 6))
display_value = f"{visible_part}{masked_part}" if len(existing_value) > 6 else existing_value
elif is_announce_url and existing_value:
# For announce_urls, show the first 10 chars and last 6 chars with * in between
if len(existing_value) > 20: # Only mask if long enough
visible_prefix = existing_value[:15]
visible_suffix = existing_value[-6:]
masked_length = len(existing_value) - 16
masked_part = "*" * min(masked_length, 15) # Limit number of asterisks
display_value = f"{visible_prefix}...{masked_part}...{visible_suffix}"
else:
display_value = existing_value
display = f"{prompt} [existing: {display_value}]"
# Show default if available
if default and existing_value is None:
display = f"{display} [default: {default}]"
display = f"{display}: "
# Prompt for input
value = input(display)
# Use existing value if user just pressed Enter and we have an existing value
if value == "" and existing_value is not None:
return existing_value
# Use default if no input and no existing value
if value == "" and default:
return default
return value
def configure_default_section(existing_defaults, example_defaults, config_comments, quick_setup=False):
"""
Helper to configure the DEFAULT section.
Returns a dict with the configured DEFAULT values.
"""
print("\n====== DEFAULT CONFIGURATION ======")
print("\n[i] Press enter to accept the default values/skip, or input your own values.")
config_defaults = {}
# Settings that should only be prompted if a parent setting has a specific value
linked_settings = {
"update_notification": {
"condition": lambda value: value.lower() == "true",
"settings": ["verbose_notification"]
},
"tone_map": {
"condition": lambda value: value.lower() == "true",
"settings": ["algorithm", "desat", "tonemapped_header"]
},
"add_logo": {
"condition": lambda value: value.lower() == "true",
"settings": ["logo_size", "logo_language"]
},
"frame_overlay": {
"condition": lambda value: value.lower() == "true",
"settings": ["overlay_text_size"]
},
"multiScreens": {
"condition": lambda value: (value.isdigit() and int(value) > 0),
"settings": ["pack_thumb_size", "charLimit", "fileLimit", "processLimit", ]
},
"get_bluray_info": {
"condition": lambda value: value.lower() == "true",
"settings": ["add_bluray_link", "use_bluray_images", "bluray_image_size", "bluray_score", "bluray_single_score"]
}
}
# Store which settings should be skipped based on linked settings
skip_settings = set()
# If this is a fresh config (no existing defaults), offer quick setup
do_quick_setup = False
if quick_setup:
do_quick_setup = input("\n[i] Do you want to quick setup with just essential settings? (y/N): ").lower() == "y"
if do_quick_setup:
print("[i] Quick setup selected. You'll only be prompted for essential settings.")
# Define essential settings for quick setup mode
essential_settings = [
"tmdb_api"
]
for key, default_value in example_defaults.items():
if key in ["default_torrent_client"]:
continue
# Skip if this setting should be skipped based on linked settings
if key in skip_settings:
# Copy default value from example config
config_defaults[key] = default_value
continue
# Skip non-essential settings in quick setup mode
if do_quick_setup and key not in essential_settings:
config_defaults[key] = default_value
continue
if key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[key]))
if isinstance(default_value, bool):
default_str = str(default_value)
existing_value = str(existing_defaults.get(key, default_value))
value = get_user_input(f"Setting '{key}'? (True/False)",
default=default_str,
existing_value=existing_value)
config_defaults[key] = value
# Check if this is a linked setting that controls other settings
if key in linked_settings:
linked_group = linked_settings[key]
# If the condition is not met, add all linked settings to the skip list
if not linked_group["condition"](value):
print(f"[i] Skipping {key}-related settings since {key} is {value}")
skip_settings.update(linked_group["settings"])
else:
is_password = key in ["api_key", "passkey", "rss_key", "tvdb_token", "tmdb_api", "tvdb_api", "btn_api"] or "password" in key.lower() or key.endswith("_key") or key.endswith("_api") or key.endswith("_url")
config_defaults[key] = get_user_input(
f"Setting '{key}'",
default=str(default_value),
is_password=is_password,
existing_value=existing_defaults.get(key)
)
if key in linked_settings:
linked_group = linked_settings[key]
if not linked_group["condition"](config_defaults[key]):
print(f"[i] Skipping {key}-related settings since {key} is {config_defaults[key]}")
skip_settings.update(linked_group["settings"])
if do_quick_setup:
get_img_host(config_defaults, existing_defaults, example_defaults, config_comments)
print("\n[i] Applied default values from example config for non-essential settings.")
return config_defaults
# Process image hosts
def get_img_host(config_defaults, existing_defaults, example_defaults, config_comments):
img_host_api_map = {
"imgbb": "imgbb_api",
"ptpimg": "ptpimg_api",
"lensdump": "lensdump_api",
"ptscreens": "ptscreens_api",
"onlyimage": "onlyimage_api",
"dalexni": "dalexni_api",
"ziplinestudio": ["zipline_url", "zipline_api_key"],
"passtheimage": "passtheima_ge_api",
"imgbox": None,
"pixhost": None
}
print("\n==== IMAGE HOST CONFIGURATION ====")
print("[i] Available image hosts: " + ", ".join(img_host_api_map.keys()))
print("[i] Note: imgbox and pixhost don't require API keys")
# Get existing image hosts if available
existing_hosts = []
for i in range(1, 11):
key = f"image_host_{i}"
if key in existing_defaults and existing_defaults[key]:
existing_hosts.append(existing_defaults[key].strip().lower())
if existing_hosts:
print(f"\n[i] Your existing image hosts: {', '.join(existing_hosts)}")
try:
default_count = len(existing_hosts) if existing_hosts else 1
number_hosts = int(input(f"\n[i] How many image hosts would you like to configure? (1-10) [default: {default_count}]: ") or default_count)
number_hosts = max(1, min(10, number_hosts)) # Limit between 1 and 10
except ValueError:
print(f"[!] Invalid input. Defaulting to {default_count} image host(s).")
number_hosts = default_count
# Ask for each image host in sequence
for i in range(1, number_hosts + 1):
# Get existing value for this position if available
existing_host = existing_hosts[i-1] if i <= len(existing_hosts) else None
existing_display = f" [existing: {existing_host}]" if existing_host else ""
valid_host = False
while not valid_host:
host_input = input(f"\n[i] Enter image host #{i}{existing_display} (e.g., ptpimg, imgbb, imgbox): ").strip().lower()
if host_input == "" and existing_host:
host_input = existing_host
if host_input in img_host_api_map:
valid_host = True
host_key = f"image_host_{i}"
config_defaults[host_key] = host_input
# Configure API key(s) for this host, if needed
api_keys = img_host_api_map.get(host_input)
if api_keys is None:
print(f"[i] {host_input} doesn't require an API key.")
continue
# Convert single string to list for consistent handling
if isinstance(api_keys, str):
api_keys = [api_keys]
# Process each key for this host
for api_key in api_keys:
if api_key in example_defaults:
if api_key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[api_key]))
is_password = api_key.endswith("_url") or api_key.endswith("_key") or api_key.endswith("_api")
config_defaults[api_key] = get_user_input(
f"Setting '{api_key}' for {host_input}",
default=str(example_defaults.get(api_key, "")),
is_password=is_password,
existing_value=existing_defaults.get(api_key)
)
else:
print(f"[!] Invalid host: {host_input}. Available hosts: {', '.join(img_host_api_map.keys())}")
# Set unused image host API keys to empty string
for host, api_key_item in img_host_api_map.items():
if api_key_item is None:
# Skip hosts that don't need API keys
continue
if isinstance(api_key_item, str):
api_keys = [api_key_item]
else:
api_keys = api_key_item
for api_key in api_keys:
if api_key in example_defaults and api_key not in config_defaults:
config_defaults[api_key] = ""
def configure_trackers(existing_trackers, example_trackers, config_comments):
"""
Helper to configure the TRACKERS section.
Returns a dict with the configured trackers.
"""
print("\n====== TRACKERS ======")
# Get list of trackers to configure
example_tracker_list = [
t for t in example_trackers
if t != "default_trackers" and isinstance(example_trackers[t], dict)
]
if example_tracker_list:
print(f"[i] Available trackers in example config: \n{', '.join(example_tracker_list)}")
print("\n[i] (default trackers list) Only add the trackers you want to upload to on a regular basis.")
print("[i] You can add other tracker configs later if needed.")
existing_tracker_list = existing_trackers.get("default_trackers", "").split(",") if existing_trackers.get("default_trackers") else []
existing_tracker_list = [t.strip() for t in existing_tracker_list if t.strip()]
existing_trackers_str = ", ".join(existing_tracker_list)
trackers_input = get_user_input(
"\nEnter tracker acronyms separated by commas (e.g. BHD, PTP, AITHER)",
existing_value=existing_trackers_str
).upper()
trackers_list = [t.strip().upper() for t in trackers_input.split(",") if t.strip()]
trackers_config = {"default_trackers": ", ".join(trackers_list)}
# Ask if user wants to update all trackers or specific ones
update_all = input("\n[i] Do you want to update ALL trackers in your default trackers list? (Y/n): ").lower() != "n"
if not update_all:
# Ask which specific trackers to update
update_specific = input("\nEnter tracker acronyms to update (comma separated), or leave blank to skip all: ").upper()
update_trackers_list = [t.strip() for t in update_specific.split(",") if t.strip()]
else:
# Update all trackers in the list
update_trackers_list = trackers_list.copy()
# Only update trackers in the update list
for tracker in trackers_list:
# Skip if not in update list (unless updating all)
if not update_all and tracker not in update_trackers_list:
print(f"\nSkipping configuration for {tracker}")
# Copy existing config if available
if tracker in existing_trackers:
trackers_config[tracker] = existing_trackers[tracker]
continue
print(f"\n\nConfiguring **{tracker}**:")
existing_tracker_config = existing_trackers.get(tracker, {})
example_tracker = example_trackers.get(tracker, {})
tracker_config = {}
if example_tracker and isinstance(example_tracker, dict):
for key, default_value in example_tracker.items():
# Skip keys that should not be prompted
if tracker == "HDT" and key == "announce_url":
tracker_config[key] = example_tracker[key]
continue
comment_key = f"TRACKERS.{tracker}.{key}"
if comment_key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[comment_key]))
if isinstance(default_value, bool):
default_str = str(default_value)
existing_value = str(existing_tracker_config.get(key, default_value))
value = get_user_input(f"Tracker setting '{key}'? (True/False)",
default=default_str,
existing_value=existing_value)
tracker_config[key] = value
else:
is_password = key in ["api_key", "passkey", "rss_key", "password", "opt_uri"] or key.endswith("rss_key")
is_announce_url = key.endswith("announce_url")
tracker_config[key] = get_user_input(
f"Tracker setting '{key}'",
default=str(default_value) if default_value else "",
is_password=is_password,
is_announce_url=is_announce_url,
existing_value=existing_tracker_config.get(key)
)
else:
print(f"[!] No example config found for tracker '{tracker}'.")
trackers_config[tracker] = tracker_config
# Offer to add more trackers from the example config
remaining_trackers = [t for t in example_tracker_list if t.upper() not in [x.upper() for x in trackers_list]]
if remaining_trackers:
print("\n[i] Other trackers available in the example config that are not in your default list:")
print(", ".join(remaining_trackers))
print("\n[i] This just adds the tracker config, not to your list of default trackers.")
print("\nFor example so you can use with -tk.")
add_more = get_user_input(
"\nEnter any additional tracker acronyms to add (comma separated), or leave blank to skip"
)
additional = [t.strip().upper() for t in add_more.split(",") if t.strip()]
for tracker in additional:
if tracker in trackers_config:
continue # Already configured
print(f"\n\nConfiguring **{tracker}**:")
example_tracker = example_trackers.get(tracker, {})
tracker_config = {}
if example_tracker and isinstance(example_tracker, dict):
for key, default_value in example_tracker.items():
if tracker == "HDT" and key == "announce_url":
tracker_config[key] = example_tracker[key]
continue
comment_key = f"TRACKERS.{tracker}.{key}"
if comment_key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[comment_key]))
if isinstance(default_value, bool):
default_str = str(default_value)
value = get_user_input(f"Tracker setting '{key}'? (True/False)",
default=default_str)
tracker_config[key] = value
else:
is_password = key in ["api_key", "passkey", "rss_key", "password", "opt_uri"] or key.endswith("rss_key")
is_announce_url = key.endswith("announce_url")
tracker_config[key] = get_user_input(
f"Tracker setting '{key}'",
default=str(default_value) if default_value else "",
is_password=is_password,
is_announce_url=is_announce_url
)
else:
print(f"[!] No example config found for tracker '{tracker}'.")
trackers_config[tracker] = tracker_config
return trackers_config
def configure_torrent_clients(existing_clients=None, example_clients=None, default_client_name=None, config_comments=None):
"""
Helper to configure the TORRENT_CLIENTS section.
Returns a dict with the configured client(s) and the selected default client name.
"""
config_clients = {}
existing_clients = existing_clients or {}
example_clients = example_clients or {}
config_comments = config_comments or {}
# Only use default_client_name if provided and in existing_clients
if default_client_name and default_client_name in existing_clients:
keep_existing_client = input(f"\nDo you want to keep the existing client '{default_client_name}'? (y/n): ").lower() == "y"
if not keep_existing_client:
print("What client do you want to use instead?")
print("Available clients in example config:")
for client_name in example_clients:
print(f" - {client_name}")
new_client = get_user_input("Enter the name of the torrent client to use",
default="qbittorrent",
existing_value=default_client_name)
default_client_name = new_client
else:
# No default client specified or not in existing_clients, ask user to select one
print("No default client found. Let's configure one.")
print("What client do you want to use?")
print("Available clients in example config:")
for client_name in example_clients:
print(f" - {client_name}")
default_client_name = get_user_input("Enter the name of the torrent client to use",
default="qbittorrent")
# Configure the default client
print(f"\nConfiguring default client: {default_client_name}")
config_clients = configure_single_client(default_client_name, existing_clients, example_clients, config_clients, config_comments)
# After configuring the default client, ask if the user wants to add additional clients
while True:
add_another = input("\n\n[i] Do you want to add configuration for another torrent client? (y/N): ").lower() == "y"
if not add_another:
break
# Show available clients not yet configured
available_clients = [c for c in example_clients if c not in config_clients]
if not available_clients:
print("All available clients from the example config have been configured.")
break
print("\nAvailable clients to configure:")
for client_name in available_clients:
print(f" - {client_name}")
additional_client = get_user_input("Enter the name of the torrent client to configure")
if not additional_client:
print("No client name provided, skipping additional client configuration.")
continue
if additional_client in config_clients:
print(f"Client '{additional_client}' is already configured.")
continue
if additional_client not in example_clients:
print(f"Client '{additional_client}' not found in example config. Available clients: {', '.join(available_clients)}")
continue
# Configure the additional client
print(f"\nConfiguring additional client: {additional_client}")
config_clients = configure_single_client(additional_client, existing_clients, example_clients, config_clients, config_comments)
return config_clients, default_client_name
def configure_single_client(client_name, existing_clients, example_clients, config_clients, config_comments):
"""Helper function to configure a single torrent client"""
# Use existing config for the selected client if present, else use example config
existing_client_config = existing_clients.get(client_name, {})
example_client_config = example_clients.get(client_name, {})
if not example_client_config:
print(f"[!] No example config found for client '{client_name}'.")
if existing_client_config:
print(f"[i] Using existing config for '{client_name}'")
config_clients[client_name] = existing_client_config
return config_clients
# Set the client type from the example config
client_type = example_client_config.get("torrent_client", client_name)
client_config = {"torrent_client": client_type}
# Process all other client settings
for key, default_value in example_client_config.items():
# this is never edited
if key == "torrent_client":
continue
comment_key = f"TORRENT_CLIENTS.{client_name}.{key}"
if comment_key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[comment_key]))
elif key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[key]))
if isinstance(default_value, bool):
default_str = str(default_value)
existing_value = str(existing_client_config.get(key, default_value))
value = get_user_input(f"Client setting '{key}'? (True/False)",
default=default_str,
existing_value=existing_value)
client_config[key] = value
else:
is_password = key.endswith("pass") or key.endswith("password")
client_config[key] = get_user_input(
f"Client setting '{key}'",
default=str(default_value) if default_value is not None else "",
is_password=is_password,
existing_value=existing_client_config.get(key)
)
config_clients[client_name] = client_config
return config_clients
def configure_discord(existing_discord, example_discord, config_comments):
"""
Helper to configure the DISCORD section.
Returns a dict with the configured Discord settings.
"""
print("\n====== DISCORD CONFIGURATION ======")
print("[i] Configure Discord bot settings for upload notifications")
discord_config = {}
existing_use_discord = existing_discord.get("use_discord", False)
enable_discord = get_user_input(
"Enable Discord bot functionality? (True/False)",
default="False",
existing_value=str(existing_use_discord)
)
discord_config["use_discord"] = enable_discord
# If Discord is disabled, set defaults and return
if enable_discord.lower() != "true":
print("[i] Discord disabled. Setting default values for other Discord settings.")
for key, default_value in example_discord.items():
if key != "use_discord":
discord_config[key] = default_value
return discord_config
# Configure other Discord settings if enabled
for key, default_value in example_discord.items():
if key == "use_discord":
continue
comment_key = f"DISCORD.{key}"
if comment_key in config_comments:
print("\n[i] " + "\n[i] ".join(config_comments[comment_key]))
if isinstance(default_value, bool):
default_str = str(default_value)
existing_value = str(existing_discord.get(key, default_value))
value = get_user_input(
f"Discord setting '{key}'? (True/False)",
default=default_str,
existing_value=existing_value
)
discord_config[key] = value
else:
is_password = key in ["discord_bot_token"]
discord_config[key] = get_user_input(
f"Discord setting '{key}'",
default=str(default_value) if default_value else "",
is_password=is_password,
existing_value=existing_discord.get(key)
)
return discord_config
def generate_config_file(config_data, existing_path=None):
"""Generate the config.py file from the config dictionary"""
# Create output directory if it doesn't exist
os.makedirs("data", exist_ok=True)
# Determine the output path
if existing_path:
config_path = existing_path
backup_path = Path(f"{existing_path}.bak")
# Create backup of existing config
if existing_path.exists():
with open(existing_path, "r", encoding="utf-8") as src:
with open(backup_path, "w", encoding="utf-8") as dst:
dst.write(src.read())
print(f"\n[✓] Created backup of existing config at {backup_path}")
else:
config_path = Path("data/config.py")
backup_path = Path("data/config.py.bak")
if config_path.exists():
overwrite = input(f"{config_path} already exists. Overwrite? (y/n): ").lower()
if overwrite != "y":
with open(config_path, "r", encoding="utf-8") as src:
with open(backup_path, "w", encoding="utf-8") as dst:
dst.write(src.read())
print(f"\n[✓] Created backup of existing config at {backup_path}")
# Convert boolean values in config to proper Python booleans
def format_config(obj):
if isinstance(obj, dict):
# Process each key-value pair in dictionaries
return {k: format_config(v) for k, v in obj.items()}
elif isinstance(obj, list):
# Process each item in lists
return [format_config(item) for item in obj]
elif isinstance(obj, str):
# Convert string "true"/"false" to Python True/False
if obj.lower() == "true":
return True
elif obj.lower() == "false":
return False
# Return unchanged for other types
return obj
# Format config with proper Python booleans
formatted_config = format_config(config_data)
# Generate the config file with properly formatted Python syntax
with open(config_path, "w", encoding="utf-8") as file:
file.write("config = {\n")
# Custom formatting function to create Python dict with trailing commas
def write_dict(d, indent_level=1):
indent = " " * indent_level
for key, value in d.items():
file.write(f"{indent}{json.dumps(key)}: ")
if isinstance(value, dict):
file.write("{\n")
write_dict(value, indent_level + 1)
file.write(f"{indent}}},\n")
elif isinstance(value, bool):
# Ensure booleans are capitalized
file.write(f"{str(value).capitalize()},\n")
else:
# Other values with trailing comma
file.write(f"{json.dumps(value, ensure_ascii=False)},\n")
write_dict(formatted_config)
file.write("}\n")
print(f"\n[✓] Configuration file created at {config_path}")
return True
if __name__ == "__main__":
print("\nUpload Assistant Configuration Generator")
print("========================================")
# Get example configuration structure first
example_config, config_comments = read_example_config()
# Try to load existing config
existing_config, existing_path = load_existing_config()
if existing_config and example_config:
just_updating = input("\nExisting config found. Are you just updating to grab any new UA config options? (Y/n): ").lower()
if just_updating == "n":
use_existing = input("\nWould you like to edit existing instead of starting fresh? (Y/n): ").lower()
if use_existing == "n":
print("\n[i] Starting with fresh configuration.")
print("Enter to accept the default values/skip, or enter your own values.")
config_data = {}
# DEFAULT section
example_defaults = example_config.get("DEFAULT", {})
config_data["DEFAULT"] = configure_default_section({}, example_defaults, config_comments, quick_setup=True)
# Set default client name if not set
config_data["DEFAULT"]["default_torrent_client"] = config_data["DEFAULT"].get("default_torrent_client", "qbittorrent")
# TRACKERS section
example_trackers = example_config.get("TRACKERS", {})
config_data["TRACKERS"] = configure_trackers({}, example_trackers, config_comments)
# TORRENT_CLIENTS section
example_clients = example_config.get("TORRENT_CLIENTS", {})
default_client = None
client_configs, default_client = configure_torrent_clients(
{}, example_clients, default_client, config_comments
)
config_data["TORRENT_CLIENTS"] = client_configs
config_data["DEFAULT"]["default_torrent_client"] = default_client
example_discord = example_config.get("DISCORD", {})
config_data["DISCORD"] = configure_discord({}, example_discord, config_comments)
generate_config_file(config_data)
else:
print("\n[i] Using existing configuration as a template.")
print("[i] Existing config will be renamed config.py.bak.")
print("[i] Press enter to accept the default values/skip, or input your own values.")
# Check for unexpected keys in existing config
existing_config = validate_config(existing_config, example_config)
# Start with the existing config
config_data = existing_config.copy()
# Ask about updating each main section separately
print("\n\n[i] Lets work on one section at a time.")
print()
# DEFAULT section
update_default = input("Do you want to update something in the DEFAULT section? (y/n): ").lower() == "y"
if update_default:
existing_defaults = existing_config.get("DEFAULT", {})
example_defaults = example_config.get("DEFAULT", {})
config_data["DEFAULT"] = configure_default_section(existing_defaults, example_defaults, config_comments)
# Set default client name (if needed)
config_data["DEFAULT"]["default_torrent_client"] = config_data["DEFAULT"].get("default_torrent_client", "qbittorrent")
else:
print("[i] Keeping existing DEFAULT section")
print()
# TRACKERS section
update_trackers = input("Do you want to update something in thee TRACKERS section? (y/n): ").lower() == "y"
if update_trackers:
existing_trackers = existing_config.get("TRACKERS", {})
example_trackers = example_config.get("TRACKERS", {})
config_data["TRACKERS"] = configure_trackers(existing_trackers, example_trackers, config_comments)
else:
print("[i] Keeping existing TRACKERS section")
print()
# TORRENT_CLIENTS section
update_clients = input("\nDo you want to update something in the TORRENT_CLIENTS section? (y/n): ").lower() == "y"
if update_clients:
print("\n====== TORRENT CLIENT ======")
existing_clients = existing_config.get("TORRENT_CLIENTS", {})
example_clients = example_config.get("TORRENT_CLIENTS", {})
default_client = config_data["DEFAULT"].get("default_torrent_client", None)
# Get updated client config and default client name
client_configs, default_client = configure_torrent_clients(
existing_clients, example_clients, default_client, config_comments
)
# Update client configs and default client name
config_data["TORRENT_CLIENTS"] = client_configs
config_data["DEFAULT"]["default_torrent_client"] = default_client
else:
print("[i] Keeping existing TORRENT_CLIENTS section")
print()
# DISCORD section update
update_discord = input("Do you want to update something in the DISCORD section? (y/n): ").lower() == "y"
if update_discord:
existing_discord = existing_config.get("DISCORD", {})
example_discord = example_config.get("DISCORD", {})
config_data["DISCORD"] = configure_discord(existing_discord, example_discord, config_comments)
else:
print("[i] Keeping existing DISCORD section")
print()
missing_discord_keys = []
missing_default_keys = []
if "DEFAULT" in example_config and "DEFAULT" in config_data:
def find_missing_default_keys(example_section, existing_section, path=""):
for key in example_section:
if key not in existing_section:
missing_default_keys.append(key)
find_missing_default_keys(example_config["DEFAULT"], config_data["DEFAULT"])
if missing_default_keys:
print("\n\n[!] Your existing config is missing these keys from example-config:")
# Only prompt for the missing keys
missing_defaults = {k: example_config["DEFAULT"][k] for k in missing_default_keys}
# Use empty dict for existing values so only defaults are shown
added_defaults = configure_default_section({}, missing_defaults, config_comments)
config_data["DEFAULT"].update(added_defaults)
if "DISCORD" in example_config:
if "DISCORD" not in config_data:
# Entire DISCORD section is missing
print("\n[!] DISCORD section is missing from your config")
add_discord = input("Do you want to add Discord configuration? (y/n): ").lower() == "y"
if add_discord:
example_discord = example_config.get("DISCORD", {})
config_data["DISCORD"] = configure_discord({}, example_discord, config_comments)
else:
config_data["DISCORD"] = example_config["DISCORD"].copy()
else:
# Check for missing keys within DISCORD section
def find_missing_discord_keys(example_section, existing_section):
for key in example_section:
if key not in existing_section:
missing_discord_keys.append(key)
find_missing_discord_keys(example_config["DISCORD"], config_data["DISCORD"])
if missing_discord_keys:
print(f"\n[!] Your DISCORD config is missing these keys: {', '.join(missing_discord_keys)}")
add_missing_discord = input("Do you want to configure the missing Discord settings? (y/n): ").lower() == "y"
if add_missing_discord:
missing_discord_config = {k: example_config["DISCORD"][k] for k in missing_discord_keys}
added_discord = configure_discord({}, missing_discord_config, config_comments)
config_data["DISCORD"].update(added_discord)
else:
for key in missing_discord_keys:
config_data["DISCORD"][key] = example_config["DISCORD"][key]
# Generate the updated config file
generate_config_file(config_data, existing_path)
else:
existing_config = validate_config(existing_config, example_config)
config_data = existing_config.copy()
missing_default_keys = []
if "DEFAULT" in example_config and "DEFAULT" in config_data:
def find_missing_default_keys(example_section, existing_section, path=""):