forked from mckrd/py_repo_beginner
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
30 lines (28 loc) · 1.09 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import time
from datetime import datetime as dt
HOST_PATH = r"C:\Windows\System32\drivers\etc\hosts"
hosts_temp = 'hosts'
redirect = "127.0.0.1"
website_list = ["www.facebook.com", "facebook.com",
"www.youtube.com", "youtube.com"]
while True:
if dt(dt.now().year, dt.now().month, dt.now().day, 8) < dt.now() < dt(dt.now().year, dt.now().month, dt.now().day, 17):
print("Working hours...")
with open(HOST_PATH, 'r+') as file:
content = file.read()
for website in website_list:
if website in content:
pass
else:
file.write(redirect+" " + website+"\n")
else:
with open(HOST_PATH, 'r+') as file:
content = file.readlines()
file.seek(0)
for line in content:
if not any(website in line for website in website_list):
file.write(line)
file.truncate()
print(
f"Not in Working Hour: {dt.now().year, dt.now().month, dt.now().day}")
time.sleep(5)