Add files via upload

Author: RohiniRG

Pomodoro_GUI/Pomodoro_gui.py

@@ -7,18 +7,13 @@
 pygame.mixer.init()
 pomo_count = 0
 break_count = 0
+enable = 0
 
 # path of host file in windows
 host_path = r"C:\Windows\System32\drivers\etc\hosts"
 
 # URL of websites to block
-block_list = [
-    'www.facebook.com', 'facebook.com',
-    'www.youtube.com', 'youtube.com',
-    'www.gmail.com', 'gmail.com',
-    'www.instagram.com', 'instagram.com',
-    'www.twitter.com', 'twitter.com'
-]
+block_list = []
 
 # redirecting above URLs to this localhost to ensure blocking
 redirect = "127.0.0.1"
@@ -29,7 +24,12 @@ def block_websites():
     the file if it is not already present and redirect it to the localhost
     for blocking
     """
-
+    global web_var
+    global enable
+    global block_list
+    global host_path
+    url = web_var.get()
+    block_list.append(url)
     try:
         # Opening the host file in reading and writing mode
         with open(host_path, 'r+') as h_file:
@@ -45,48 +45,103 @@ def block_websites():
                 else:
                     h_file.write(redirect + "\t" + website + "\n")
 
+        tk.messagebox.showinfo("Blocked", f"{url} successfully blocked!")
+        enable = 1
+        web_var.set("")
+
     except PermissionError:
-        tk.messagebox.showinfo("Error", "Run cmd in the admin mode and then try again!\nDeselect the option to prevent this popup to show again.")
+        tk.messagebox.showinfo("Error", "Run cmd in the admin mode and then try again!")
+        web_var.set("")
 
     except (FileNotFoundError, NameError):
-        tk.messagebox.showinfo("Error", "Functionality not supported in your OS!\nDeselect the option to prevent this popup to show again.")
+        tk.messagebox.showinfo("Error", "Functionality not supported in your OS!")
+        web_var.set("")
 
 
 def remove_websites():
     """
     The function will unblock the block_list websites by opening the file
     and removing the changes we made before
     """
+    global block_list
+    global host_path
     try:
-        with open(host_path, "r+") as file:
-            content = file.readlines()
-            file.seek(0)
-            for lines in content:
-                if not any(website in lines for website in block_list):
-                    file.write(lines)
-            file.truncate()
+        if enable:
+            # Opening the host file in reading and writing mode
+            with open(host_path, "r+") as file:
+
+                # making each line of file into a list
+                content = file.readlines()
+
+                # sets the file pointer at the beginning of the file
+                file.seek(0)
+
+                # Traversing through each line of the host file and
+                # checking for the websites to be blocked
+                for lines in content:
+                    if not any(website in lines for website in block_list):
+                        file.write(lines)
+                        
+                # Truncating the file to its original size
+                file.truncate()
+
+            block_list.clear()
+            enable = 0
     except:
         pass
     finally:
         pass
 
 
-def high_focus():
+def blocker():
+    """
+    The function asks input from user to block websites for high focus mode.
+    """
     global enable
-    if enable.get() == 1:
-        block_websites()
-    else:
-        remove_websites()
+    global popup_4
+    popup_4 = tk.Toplevel(root)
+    popup_4.title("Website Blocker!")
+    popup_4.geometry("360x220")
+    popup_4.config( bg = 'DodgerBlue4')
+
+    global block_list
+    global web_var
+    web_var=tk.StringVar()
+
+    pass_label = tk.Label(popup_4, text = 'Enter URL to block:', font = ('Arial',12, 'bold'), bg = 'DodgerBlue4', fg = 'white')
+    pass_entry = tk.Entry(popup_4, textvariable = web_var, font = ('Arial',12, 'bold'))
+
+    sub_btn = tk.Button(popup_4, text = 'Block', font = ('Arial',12, 'bold'), command = block_websites, bg='gold', activebackground='yellow')
+
+    text_to_put = '*Supported for windows ONLY\n*You can add multiple urls\n*Don\'t forget to unblock after'
+
+    instructions = tk.Label(popup_4, text = text_to_put, font = ('Arial',12, 'bold'), justify='left', bg = 'sky blue')
+
+    unblock_btn = tk.Button(popup_4, text = 'Unblock all', font = ('Arial',12, 'bold'), command = remove_websites, state='disabled', width = 23, height = 2, bg='gold', activebackground='yellow')
+
+    if enable:
+        unblock_btn.config(state='normal')
+
+    pass_label.place(x=25, y=10)
+    pass_entry.place(x=25, y=34)
+    sub_btn.place(x=255, y=30)
+    instructions.place(x=25, y=80)
+    unblock_btn.place(x=50, y=150)
 
 
 def break_timer():
+    """
+    5 min timer popup window acting as a callback function to the break timer button
+    """
+    global enable
     global popup_2
     popup_2 = tk.Toplevel(root)
     popup_2.title("Break Timer!")
     popup_2.geometry("370x120")
     round = 0
 
     try:
+        # Creating a continous loop of text of time on the screen for 25 mins
         t = 5*60
         while t>-1:
             minute_count = t // 60
@@ -99,18 +154,22 @@ def break_timer():
             t -= 1
     except:
         pass    
-
-    pygame.mixer.music.load("beep.wav")
-    pygame.mixer.music.play(loops=0)
-
+    
+    # Setting up an alarm sound and popup window to let user know when the time is up
     if t == -1:
         tk.messagebox.showinfo("Time's up!", "Break is over!\nTime to get to work!")
         popup_2.destroy()
         global break_count
+        pygame.mixer.music.load("./Pomodoro_GUI/beep.wav")
+        pygame.mixer.music.play(loops=1)
         break_count += 1
 
 
 def show_report():
+    """
+    The function acts as a callback for show report button and shows the report the hours 
+    of work they have put in. 
+    """
     global popup_3
     popup_3 = tk.Toplevel(root)
     popup_3.title("Report")
@@ -126,13 +185,17 @@ def show_report():
 
 
 def pomodoro_timer():
+    """
+    25 min timer popup window acting as a callback function to the work timer button
+    """
     global popup_1
     popup_1 = tk.Toplevel(root)
     popup_1.title("Work Timer!")
     popup_1.geometry("370x120")
     round = 0
 
     try:
+        # Creating a continous loop of text of time on the screen for 25 mins
         t = 25*60
         while t>-1:
             minute_count = t // 60
@@ -145,38 +208,39 @@ def pomodoro_timer():
             t -= 1
     except:
         pass    
-
-    pygame.mixer.music.load("beep.wav")
-    pygame.mixer.music.play(loops=1)
-
+    
+    # Setting up an alarm sound and popup window to let user know when the time is up
     if t == -1:
         tk.messagebox.showinfo("Time's up!", "Pomodoro completed successfully!\nYou deserve a break!")
         popup_1.destroy()
         global pomo_count
         pomo_count += 1
+        pygame.mixer.music.load("./Pomodoro_GUI/beep.wav")
+        pygame.mixer.music.play(loops=0)
 
 
 def main():
+    """
+    This function produces the main screen of the Pomodoro timer with options to 
+    select the 25mins work timer, 5mins break timer, block websites for extra focus and 
+    another option to see the statistics of the time you've put in the work
+    """
+    # Creating the root window (main screen)
     global root
     root = tk.Tk()
     root.title('Timer')
     root.geometry('470x608')
 
-    bg = tk.PhotoImage(file = "bg.png")
-    
-    # Show image using label
+    # Setting the screen background
+    bg = tk.PhotoImage(file = "./Pomodoro_GUI/bg.png")
     label1 = tk.Label( root, image = bg)
     label1.place(x = 0, y = 0)
 
-    global count
-
     intro1 = tk.Label(root, text = 'POMODORO TIMER', bg = 'snow', fg = 'maroon', font = ('Arial', 25, 'bold'))
-    intro1.place(x=100, y=120)
+    intro1.place(x=100, y=100)
 
-    global enable
-    enable = tk.IntVar()
-    check  = tk.Checkbutton(root, text = 'Enable website blocker', variable = enable, font = ('Arial', 12, 'bold'), bg='gold', activebackground='yellow', height = 1, width = 25, onvalue=1, offvalue=0, command=high_focus)
-    check.place(x=100, y=190)
+    blocker_btn = tk.Button(root, text = 'WEBSITE BLOCKER', command = blocker, font = ('Arial', 12, 'bold'), bg='gold', activebackground='yellow', height = 3, width = 25)
+    blocker_btn.place(x=100, y=150)
 
     start_btn = tk.Button(root, text = 'START WORK TIMER', command = pomodoro_timer, font = ('Arial', 12, 'bold'), bg='gold', activebackground='yellow', height = 3, width = 25)
     start_btn.place(x=100, y=250)

Pomodoro_GUI/README.md

@@ -16,7 +16,7 @@
 ```
 
 ## Working:
-![GIF](https://media.giphy.com/media/Z8k8V5FRALLMLKsOpJ/giphy.gif)
+![GIF](https://media.giphy.com/media/rFugmcue93IERozmWw/giphy.gif)
 
 
 ## Author: