c2vi/dotfiles
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more yt_block bug fixes

Browse Source
This commit is contained in:
Sebastian Moser
2024-07-21 23:21:21 +02:00
parent 31aabfe08b
commit 661ef90165
1 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
scripts/yt-block/main.py
View File

@@ -62,8 +62,8 @@ def cmd_guard():
if state["yt_time_current"] == 0: if state["yt_time_current"] == 0:
block_yt() block_yt()
# if time_current in state is 0, block yt # if time_current in state is greater than 0, unblock yt
if state["yt_time_current"] >= 0: if state["yt_time_current"] > 0:
unblock_yt() unblock_yt()
# decrement time_current # decrement time_current
@@ -97,8 +97,6 @@ def cmd_info():
state = read_state(pwd) state = read_state(pwd)
kill_mc()
print("YouTube") print("YouTube")
print("time_left:", state["yt_time_left"]) print("time_left:", state["yt_time_left"])
print("time_current:", state["yt_time_current"]) print("time_current:", state["yt_time_current"])
@@ -157,7 +155,7 @@ def get_hosts():
continue continue
split = line.split(" ") split = line.split(" ")
try: try:
hosts.append([split[1].strip(), split[0].strip()]) hosts.append([split[0].strip(), split[1].strip()])
except: except:
print("error with geting hosts from /etc/hosts: ", split) print("error with geting hosts from /etc/hosts: ", split)
return hosts return hosts
@@ -168,7 +166,7 @@ def write_hosts(hosts):
with open("/etc/hosts", "w") as file: with open("/etc/hosts", "w") as file:
lines = [] lines = []
for entry in hosts: for entry in hosts:
lines.append(entry[1] + " " + entry[0]) lines.append(entry[0] + " " + entry[1])
file.write("\n".join(lines) + "\n") file.write("\n".join(lines) + "\n")
@@ -176,29 +174,37 @@ def write_hosts(hosts):
def block_yt(): def block_yt():
hosts = get_hosts() hosts = get_hosts()
for entry in YT_HOSTS: for entry in YT_HOSTS:
if entry not in hosts:
hosts.append(entry) hosts.append(entry)
write_hosts(hosts) write_hosts(hosts)
os.system("iptables -I INPUT -s 188.21.9.20 -j REJECT")
def unblock_yt(): def unblock_yt():
hosts = get_hosts() hosts = get_hosts()
new_hosts = [] new_hosts = []
for entry in hosts: for entry in hosts:
if entry[0] == "youtube.com" or entry[0] == "www.youtube.com": if entry[1] == "youtube.com" or entry[1] == "www.youtube.com":
continue continue
else: else:
new_hosts.append(entry) new_hosts.append(entry)
write_hosts(new_hosts) write_hosts(new_hosts)
os.system("iptables -D INPUT 1")
def kill_mc(): def kill_mc():
try:
output = subprocess.check_output(['bash', '-c', "ps fax | grep minecraft"]) output = subprocess.check_output(['bash', '-c', "ps fax | grep minecraft"])
for line in output.decode().split("\n"): for line in output.decode().split("\n"):
if line.find("java") != -1: if line.find("java") != -1:
kill_line(line) kill_line(line)
except Exception as e:
print("killing failed", e)
def kill_line(line): def kill_line(line):
pid = int(line.split(" ")[1]) #print("line:", line)
pid = int(line.split(" ")[0])
#print("killing pid:", pid)
os.system(f"kill {pid}") os.system(f"kill {pid}")
if __name__ == "__main__": if __name__ == "__main__":