import datetime
import time
import elevate
import os


elevate.elevate()

host_path = r"C:\Windows\System32\drivers\etc\hosts" #r đằng trước string để coi như cái j viết trong string máy sẽ dùng như thế. Search raw string python

redirect = "127.0.0.1"

website_list = [] #các website để block

exe_list = ["msedge.exe", "notepad.exe"]

def CheckTime(time_start, time_end): #check xem thời gian hiện tại có trong thời gian gian không
    time_start_hour = int(time_start[:2])
    time_start_min = int(time_start[-2:])
    datetime_time_start = datetime.datetime(datetime.datetime.now().year, datetime.datetime.now().month, datetime.datetime.now().day, time_start_hour, time_start_min)

    time_end_hour = int(time_end[:2])
    time_end_min = int(time_end[-2:])
    datetime_time_end = datetime.datetime(datetime.datetime.now().year, datetime.datetime.now().month, datetime.datetime.now().day, time_end_hour, time_end_min)

    if datetime_time_start < datetime.datetime.now() < datetime_time_end:
        return True

    return False

def BlockWebsite(time_start, time_end): #block từ time_start -> time_end (2 cái này là string) vd: "18:00"

    if CheckTime(time_start, time_end):
        print("websites still blocked")

        #code block websites (viết vào file hosts)
        with open(host_path, 'r+') as file:
            file_content = file.read()
            for website in website_list:
                if website in file_content:
                    pass
                else:
                    file.write(redirect + " " + website + '\n')
    else:
        with open(host_path, "r+") as file:
            file_content = file.readlines()
            file.seek(0)
            for line in file_content:
                if not any(website in line for website in website_list):
                    file.write(line)
            file.truncate()
        #print('websites not blocked')

def BlockExe(time_start, time_end): #block từ time_start -> time_end (2 cái này là string) vd: "18:00"
    if CheckTime(time_start, time_end):
        print("exe still blocked")

        for exe in exe_list:
            cmd_string = "taskkill /f /im  " + exe
            os.system(cmd_string) # os.system là để chạy command trên cmd

time_start_global = "00:00"
time_end_global = "05:40"




if __name__ == "__main__":
    while True:
        BlockExe(time_start_global, time_end_global)
        BlockWebsite(time_start_global, time_end_global)
        time.sleep(1)