source.py

#!/usr/bin/env python3

import re
import random
import requests
import bs4
from bs4 import BeautifulSoup
import gi
gi.require_version('Notify', '0.7')
from gi.repository import Notify,GdkPixbuf
import os
import time
from multiprocessing import Process
import sys

out = open('/dev/null', 'w')
sys.stderr = sys.stdout = out

class Culture:
    # method to clear input_str of nested parentheses with stuff in between them
    # copyright: https://stackoverflow.com/a/14598135
    def clearContent(self, input_str):
        ret = ''
        skip1c = 0
        skip2c = 0
        for i in input_str:
            if i == '[':
                skip1c += 1
            elif i == '(':
                skip2c += 1
            elif i == ']' and skip1c > 0:
                skip1c -= 1
            elif i == ')'and skip2c > 0:
                skip2c -= 1
            elif skip1c == 0 and skip2c == 0:
                ret += i
        return ret

    #  get data for notification
    def getMusic(self):
        self.type = "Music curiosities:"

        #  data set for the urls used for notifications
        data_set = ["Nai", "Chitară", "Pian"]
        # TO DO: add more to dataset

        #  get a randome url source
        url = "https://ro.wikipedia.org/wiki/" + random.choice(data_set)
        source = requests.get(url).text
        bs = BeautifulSoup(source, "lxml")

        #  get the content to put in the notification
        content = bs.p.text
        content = self.clearContent(content)

        self.info = content
        return self

    #  get data for notifications
    def getWoman(self):
        self.type = "Imortant woman:"

        # data set for the urls used for notifications
        data_set = ["Marie_Curie", "Audrey_Hepburn", "Grace_Hopper", "Katherine_Johnson", "J._K._Rowling", "Margaret_Hamilton_(om_de_știință)", "Elisa_Leonida_Zamfirescu"]
        # TO DO: add more to data_set

        # get a randome url source
        url = "https://ro.wikipedia.org/wiki/" + random.choice(data_set)
        source = requests.get(url).text
        bs = BeautifulSoup(source, "lxml")

        # get the content to put in the notification
        content = bs.p.text
        content = self.clearContent(content)

        self.info = content
        return self
    
    #  get data for notifications
    def getDestinations(self):
        self.type = "Beautiful places:"

        # data set for the urls used for notifications
        data_set = ["Madrid", "Barcelona", "Roma", "Sankt_Petersburg", "Veneția", "Lisabona", "München", "Budapesta"]
        # TO DO: add more to data_set

        # get a randome url source
        url = "https://ro.wikipedia.org/wiki/" + random.choice(data_set)
        source = requests.get(url).text
        bs = BeautifulSoup(source, "lxml")

        # get the content to put in the notification
        content = bs.p.text
        content = self.clearContent(content)

        self.info = content
        return self

    # TO DO(optional update): add more notification content scrapping methods

def getCultureNotification():
    #  choose randomly the notification type
    choice = random.choice([0, 1, 2])
    
    if choice == 0 :
        return Culture().getMusic()
    if choice == 1 :
        return Culture().getWoman()
    if choice == 2 :
        return Culture().getDestinations()

    #  TO DO(optional update): don't repeat a previous shown notification


#  data for health notifications type
class Health:
    def Water(self):
        self.type = "Hidrate!"
        self.info = "It's recommended to drink a glass of water every 30 minutes."
        return self

    def Fruit(self):
        self.type = "Eat a fruit!"
        self.info = "Take your portion of Vitamin C"
        return self

    def Pause(self):
        self.type = "Break time!"
        self.info = "You shouldn't spend more than an hour on the chair!"
        return self

#  Health_Water notification module
def Health_Water():
    Notify.init("Water")

    while True:
        time.sleep(1800)  # sleep for 1800 seconds

        content = Health().Water()
        notification = Notify.Notification.new(
            content.type,
            content.info
        )
        notification.set_urgency(0)
        icon = GdkPixbuf.Pixbuf.new_from_file("./Icons/water32.png")
        notification.set_icon_from_pixbuf(icon)
        notification.show()
        file = "unconvinced.mp3"
        os.system("mpg123 " + file + " 2> /dev/null")
    
    Notify.uninit("Water")

#  Health_Fruit notification module
def Health_Fruit():
    Notify.init("Fruit")

    while True:
        time.sleep(5400)  # sleep for 5400 seconds

        content = Health().Fruit()
        notification = Notify.Notification.new(
            content.type,
            content.info
        )
        notification.set_urgency(0)
        icon = GdkPixbuf.Pixbuf.new_from_file("./Icons/fruit32.png")
        notification.set_icon_from_pixbuf(icon)
        notification.show()
        file = "unconvinced.mp3"
        os.system("mpg123 " + file + " 2> /dev/null")

    Notify.uninit("Fruit")

#  Health_Pause notification module
def Health_Pause():
    Notify.init("Pause")

    while True:
        time.sleep(3600)  # sleep for 3600 seconds

        content = Health().Pause()
        notification = Notify.Notification.new(
            content.type,
            content.info
        )
        notification.set_urgency(0)
        icon = GdkPixbuf.Pixbuf.new_from_file("./Icons/pause32.png")
        notification.set_icon_from_pixbuf(icon)
        notification.show()
        file = "unconvinced.mp3"
        os.system("mpg123 " + file + " 2> /dev/null")


    Notify.uninit("Pause")

#  Culture notification module
def Culture_notification():
    Notify.init("Culture")

    while True:   
        time.sleep(10000)  # sleep for 10000 seconds

        content = getCultureNotification()
        notification = Notify.Notification.new(
            content.type,
            content.info
        )
        notification.set_urgency(0)
        icon = GdkPixbuf.Pixbuf.new_from_file("./Icons/culture32.png")
        notification.set_icon_from_pixbuf(icon)
        notification.show()
        file = "unconvinced.mp3"
        os.system("mpg123 " + file + " 2> /dev/null")

    Notify.uninit("Culture")


def main():

    H_W_process = Process(target=Health_Water)
    H_W_process.start()
    
    H_F_process = Process(target=Health_Fruit)
    H_F_process.start()
    
    H_P_process = Process(target=Health_Pause)
    H_P_process.start()

    C_process = Process(target=Culture_notification)
    C_process.start()

    H_W_process.join()
    H_F_process.join()
    H_P_process.join()
    C_process.join()




if __name__ == "__main__":
    main()