Website_blocker.py

'''
 
This Program helps user to unblock and block websites on windows using hosts file.

'''

import platform
import os
import sys

url=''

 
# Function to check if program is running on windows os or not.
def check_w():
    if platform.system().lower() != 'windows':
       return 1
    return 0


# Function to pause program.
def pause():
   x = input('\nPress any key to continue...')
   if check_w() == 0:
      os.system('cls')
   else:
      os.system('clear')

      
# Function to get website URL as input from user.
def get_url():
    x = input('\nEnter website to block:')
    return x
 

# Function to block url given as input.
def engine(url):
 
    # Code to obtain domain from website.
    if url.startswith('https://'):
               url = url[8:]
    elif url.startswith('http://'):
               url = url[7:]
               
    
      
    # Creating batch file to block website by pointing it to local host using hosts file.
    with open('script.bat','a') as f:
          f.write('echo 127.0.0.1 '+url+' >>C:\\Windows\\System32\\drivers\\etc\\hosts \n')


# Function to import blocklisted urls from text file.
def import_text_file():
  if os.path.exists('script.bat'):
    os.remove("script.bat")
  file = input('\nEnter File:')
  if os.path.exists(file):
    with open(file) as f:
      for line in f:
        engine(line.strip())
    print('Please run script.bat generated as administrator ...')
  else:
    print('File Not Found!\n')
  pause()
  menu()
  sys.exit()
 

# Block internet.
def internetblock():
    os.system('ipconfig /release')
    os.system('cls')
    print('\nInternet is now blocked...')
    pause()
    menu()
    sys.exit()

 
# Unblock Internet
def internetunblock():
    os.system('ipconfig /renew')
    os.system('cls')
    print('\nInternet is now unblocked...')
    pause()
    menu()
    sys.exit()
 
 
# Function to delete all Blocking Rules for websites.
def unblockall():

    # Creating windows batch file to truncate hosts file.
    with open('script.bat','w') as f:
           f.write('echo #>C:\\Windows\\System32\\drivers\\etc\\hosts')
    print('Please run script.bat generated as administrator to delete all blocked websites.')
    pause()
    menu()
    sys.exit()

 
# Function to block given website.
def block():
    url = get_url()
 
    # Code to obtain domain from website.
    if url.startswith('https://'):
               url = url[8:]
    elif url.startswith('http://'):
               url = url[7:]
                 
    # Creating batch file to block website by pointing it to local host using hosts file.
    with open('script.bat','w') as f:
          f.write('echo 127.0.0.1 '+url+' >>C:\\Windows\\System32\\drivers\\etc\\hosts\n')
    print('please run script.bat generated as administrator to block website.')
    pause()
    menu()
    sys.exit()


# Function to unblock blocked website.
def unblock():
    flag = 0
    blocklist = []
    url = input('\nEnter website to unblock:')
    if url.startswith('https://'):
               url = url[8:]
    elif url.startswith('http://'):
               url = url[7:]
       
    with open('C:\\Windows\\System32\\drivers\\etc\\hosts','r') as f:
       for line in f:
           if len(line.strip())>9:
                 blocklist.append(line.split()[1])
                 
    with open('script.bat','w') as f:
           f.write('echo #>C:\\Windows\\System32\\drivers\\etc\\hosts\n')
    for i in blocklist:
       if url !=i :
          flag += 1
          with open('script.bat','a') as f:
             f.write('echo 127.0.0.1 '+i+' >>C:\\Windows\\System32\\drivers\\etc\\hosts\n')
    if flag == len(blocklist):
         print('Entered website is not blocked!')
         if os.path.exists('script.bat'):
            os.remove('script.bat')
            pause()
            menu()
            sys.exit()
             
    print('\nPlease run script.bat script as administrator to continue...')
    pause()
    menu()
    sys.exit()
       
              
# Function to display currently blocked domains.
def display():
    mark=0
    print('\n===Currently Blocked Websites===\n')
    with open('C:\\Windows\\System32\\drivers\\etc\\hosts','r') as f:
        for line in f:
            if line.startswith('#') is False and len(line.strip()) > 9:
                print(line.split()[1])
                mark=1

    if mark==0:
       print(" No Websites Blocked!")
    pause()
    menu()
    sys.exit()


# Function to import hostfile.
def import_hostfile():
  print('\nWarning: Please Ensure that you trust the hostfile you import they may contain malware\n\n')
  file=input('Enter hostfile Location to be imported:')
  if os.path.exists(file):
    with open('script.bat','w') as f:
      f.write('move \"'+os.getcwd()+'\\'+file+'\" C:\\Windows\\System32\\drivers\\etc\\hosts')
    print('\nPlease run script.bat script as administrator to continue...')
  else:
    print('File Not Found!\n')
  pause()
  menu()
  sys.exit()


# Function to export hostile to current working directory.    
def export_hostfile():
  with open('script.bat','w') as f:
     f.write('copy C:\\Windows\\System32\\drivers\\etc\\hosts \"'+os.getcwd()+'\"')
  print('\nPlease run script.bat script as administrator to continue...')
  pause()
  menu()
  sys.exit()

       
# Function to display menu.
def menu():
    if check_w() != 0:
     print('Platform not supported!\nPress any key to continue...')
     sys.exit(0)
    os.system('cls')
    print('\n\t===Simple Website Blocker===')
    print('\n\n1)Block A website')
    print('2)Unblock blocked website')
    print('3)Display blocked websites')
    print('4)Delete all Rules to block Websites')
    print('5)Block Internet Connection')
    print('6)Unblock Internet Connection')
    print('7)Import Website Blocklist from text File')
    print('8)Import Hostfile')
    print('9)Export Hostfile')
    
    x = input('\nEnter choice:')
    if x == '1':
       block()
    elif x == '2':
       unblock()
    elif x == '3':
       display()
    elif x == '4':
       unblockall()
    elif x == '5':
       internetblock()
    elif x == '6':
       internetunblock()
    elif x == '7':
       import_text_file()
    elif x == '8':
       import_hostfile()
    elif x == '9':
       export_hostfile()
    elif x.lower() == 'c' or x.lower() == 'close':
       sys.exit()
    else:
        print('Wrong choice!')
        x = input('press any key to continue...')
        menu()
 
menu()