blocker.py

#!/usr/bin/python3
import time
from datetime import datetime as dt
from platform import system as sys

from src import ip_addresses
from src import web_list

if sys == "Windows":
    host_temp = r"C:\Windows\System32\drivers\etc\hosts"
# redirect = ip_addresses
else:
    host_temp = r"/etc/hosts"
while True:
    if (dt(dt.now().year, dt.now().month, dt.now().day, 8, 59) < dt.now() < dt(dt.now().year, dt.now().month, dt.now().day, 17, 59)):
        print("Working hours...")
        with open(host_temp, mode='r+', encoding='utf-8') as f:
            content = f.read()
            # print(content)
            for website in web_list:
                if website in content:
                    # print(content)
                    pass
                else:
                    for ip_address in ip_addresses:
                        # redirect = ip_address
                        f.write(ip_address + "\t" + website + "\n")
    else:
        with open(host_temp, mode='r+', encoding='utf-8') as f:
            content = f.readlines()
            f.seek(0)
            for line in content:
                if not any(website in line for website in web_list):
                    f.write(line)
            f.truncate()
        print("Fun hours...")
        
    time.sleep(5)