55 lines
1.1 KiB
Python
55 lines
1.1 KiB
Python
|
from sys import argv
|
||
|
|
from os import path, mkdir
|
||
|
|
from shutil import rmtree
|
||
|
|
|
||
|
|
channels = argv[1:]
|
||
|
|
|
||
|
|
if not path.isdir('data'):
|
||
|
|
print("Creating data")
|
||
|
|
mkdir('data')
|
||
|
|
|
||
|
|
if path.isfile('data/channels.txt'):
|
||
|
|
file = open('data/channels.txt', 'r+')
|
||
|
|
existing = [c.strip() for c in file]
|
||
|
|
# print(existing)
|
||
|
|
else:
|
||
|
|
file = open('data/channels.txt', 'w')
|
||
|
|
existing = []
|
||
|
|
|
||
|
|
if channels == []:
|
||
|
|
if existing == []:
|
||
|
|
print("No channels")
|
||
|
|
else:
|
||
|
|
print("Channels: " + ','.join(existing))
|
||
|
|
|
||
|
|
for ch in channels:
|
||
|
|
remove = ch.startswith('-')
|
||
|
|
if remove:
|
||
|
|
ch = ch[1:]
|
||
|
|
|
||
|
|
if ch not in existing:
|
||
|
|
print(f"Channel {ch} doesn't exist")
|
||
|
|
continue
|
||
|
|
|
||
|
|
rmtree('data/' + ch)
|
||
|
|
existing.remove(ch)
|
||
|
|
|
||
|
|
file.truncate(0)
|
||
|
|
file.seek(0)
|
||
|
|
file.writelines([e + '\n' for e in existing])
|
||
|
|
|
||
|
|
print("Removed " + ch)
|
||
|
|
else:
|
||
|
|
if ch in existing:
|
||
|
|
print(f"Channel {ch} already exists")
|
||
|
|
continue
|
||
|
|
|
||
|
|
file.write(ch + '\n')
|
||
|
|
try:
|
||
|
|
mkdir('data/' + ch)
|
||
|
|
except FileExistsError:
|
||
|
|
print(f"Channel {ch} already exists but wasn't in channels.txt, now added")
|
||
|
|
|
||
|
|
existing += [ch]
|
||
|
|
print("Created " + ch)
|