app.py

import time
from datetime import datetime as dt
import os

def get_hosts_path():
    if os.name == 'posix':
        return "/etc/hosts"
    elif os.name == 'nt':
        return r"C:\Windows\System32\drivers\etc\hosts"
    else:
        raise OSError("Unsupported operating system.")


# Enter the site name which you want to block
sites_to_block = [
    "www.facebook.com",
    "facebook.com",
    "www.youtube.com",
    "youtube.com",
    "www.gmail.com",
    "gmail.com",
]

# different hosts for different os
default_hoster = get_hosts_path() #get the correct host path depending on if you're using linux or windows
redirect = "127.0.0.1"


def is_work_time(start_hour, end_hour):
    now = dt.now()
    start = dt(now.year, now.month, now.day, start_hour)
    end = dt(now.year, now.month, now.day, end_hour)

    if start_hour < end_hour:
        return start <= now < end
    else:
        return not (end <= now < start)  # handles overnight shifts

def block_websites(start_hour, end_hour):
    while True:
        try:
            if is_work_time(start_hour, end_hour):
                print("Do the work ....")
                with open(default_hoster, "r+") as hostfile:
                    hosts = hostfile.read()
                    for site in sites_to_block:
                        if site not in hosts:
                            hostfile.write(redirect + " " + site + "\n")
            else:
                with open(default_hoster, "r+") as hostfile:
                    hosts = hostfile.readlines()
                    hostfile.seek(0)
                    for host in hosts:
                        if not any(site in host for site in sites_to_block):
                            hostfile.write(host)
                    hostfile.truncate()
                print("Good Time")
            time.sleep(3)
        except PermissionError as e:
            print(f"Caught a permission error: Try Running as Admin {e}")
            # handle the error here or exit the program gracefully
            break



if __name__ == "__main__":
    block_websites(9, 21)