Added icon.png

pastafaristats/pastafariagent.py

@@ -5,6 +5,28 @@ from pastafaristats import send_info_daemon
 from time import sleep
 import os
 import psutil
+from pathlib import Path
+import signal
+
+#Get user_home and environment variables
+
+user_home=str(Path.home())
+
+path_config=user_home+'/.config/pastafari/stats.cfg'
+
+# Create configuration
+
+Path(os.path.dirname(path_config)).mkdir(parents=True, exist_ok=True)
+
+if not os.path.isfile(path_config):
+    config_data="""[DEFAULT]
+    url_server={}
+    """.format('http://localhost:5000/api_token')
+    
+    with open(path_config, 'w') as f:
+        f.write(config_data)
+
+# Get functions
 
 def quit_app():
     print('Quitting Application...')
@@ -31,12 +53,18 @@ def quit_app():
     # Killing using pid
     
     #p.stop()
+def show_config_win():
+    
+    winconfig.show()
 
 def send_data():
     print('Init daemon...')
+    
+    url_monit=send_info_daemon.load_config()
+    
     while True:
         
-        send_info_daemon.run()
+        send_info_daemon.run(url_monit)
         
         sleep(120)
     
@@ -62,6 +90,37 @@ class ProcessRunnable(QRunnable):
         del self.t
     """
     
+class WinConfig(QWidget):
+    def __init__(self, url_monit='http://192.168.1.51/monit'):
+        super().__init__()
+        
+        url_monit=send_info_daemon.load_config()
+        
+        self.text=QLabel("Change monitoritation url, example: http://192.168.1.51/monit/api_token")
+        self.input=QLineEdit(url_monit)
+        self.button=QPushButton("Save")
+        
+        self.layout=QVBoxLayout(self)
+        self.layout.addWidget(self.text)
+        self.layout.addWidget(self.input)
+        self.layout.addWidget(self.button)
+  
+        self.button.clicked.connect(self.save_config)
+        
+    
+    def save_config(self):
+        
+        config_data="""[DEFAULT]
+        url_server={}
+        """.format(self.input.text())
+        
+        with open(path_config, 'w') as f:
+            f.write(config_data)
+        
+        print('Config saved...')
+                
+        self.hide()
+  
 app = QApplication([])
 app.setQuitOnLastWindowClosed(False)
   
@@ -79,10 +138,10 @@ menu = QMenu()
 
 option1 = QAction("Pastafari monitoring")
 menu.addAction(option1)
-"""
-option2 = QAction("GFG")
+
+option2 = QAction("Setup Monitoring url")
+option2.triggered.connect(show_config_win)
 menu.addAction(option2)
-"""
   
 # To quit the app
 quit = QAction("Quit")
@@ -92,11 +151,30 @@ menu.addAction(quit)
 # Adding options to the System Tray
 tray.setContextMenu(menu)
 
+# Prepare url monitoritation
+
+winconfig=WinConfig()
+
+winconfig.resize(400, 200)
+
 # Begin background process
 
 p = ProcessRunnable(target=send_data, args=())
 p.start()
 
+# Catch sigint
+
+def catch_signal(sig, frame):
+    print('Exiting from app...')
+    pid=os.getpid()
+    
+    p=psutil.Process(pid)
+    
+    p.terminate()
+    exit(0)
+
+signal.signal(signal.SIGINT, catch_signal)
+
 # Begin tray loop
 
 app.exec_()

pastafaristats/send_info_daemon.py

@@ -14,44 +14,44 @@ import signal
 
 #url="http://url/to/info"
 
-# Get config from /etc/pastafari or ~/.config/pastafari
-
-# Load configuration
-
-yes_config=False
-
 user_home=str(Path.home())
 
-config = configparser.ConfigParser(interpolation=None)
+def load_config():
     
-if os.path.isfile(user_home+'/.config/pastafari/stats.cfg'):
+    yes_config=False
+    
+    config = configparser.ConfigParser(interpolation=None)
+
+    if os.path.isfile(user_home+'/.config/pastafari/stats.cfg'):
         
         config.read(user_home+'/.config/pastafari/stats.cfg')
         
         yes_config=True
 
-elif os.path.isfile('/etc/pastafari/stats.cfg'):
+    elif os.path.isfile('/etc/pastafari/stats.cfg'):
         
         config.read('/etc/pastafari/stats.cfg')
         
         yes_config=True
     
-    
-if not yes_config:
+    if not yes_config:
         
         exit("Sorry, cannot load config file")
 
-if not 'DEFAULT' in config:
+    if not 'DEFAULT' in config:
         
         exit("Sorry, config file need [DEFAULT] section")
 
-if not 'url_server' in config['DEFAULT']:
+    if not 'url_server' in config['DEFAULT']:
         
         exit("Sorry, config file need url_server variable in [DEFAULT] section")
         
-url=config['DEFAULT']['url_server']
+    url=config['DEFAULT']['url_server']
     
-def run():
+    return url
+
+
+def run(url):
     
     network_info=psutil.net_io_counters(pernic=False)
 
@@ -103,21 +103,31 @@ def run():
         
         print('Cannot connect to data server -> '+str(e))
         
-def start():
+        #Reload config
+        url=load_config()
+        
+
+def start(url):
+
+    # Get config from /etc/pastafari or ~/.config/pastafari
+
+    # Load configuration
 
     while True:
         
-        run()
+        run(url)
         
         sleep(120)
 
-def catch_signal(sig, frame):
+if __name__=='__main__':
+    
+    def catch_signal(sig, frame):
         print('Exiting...')
         exit(0)
 
-signal.signal(signal.SIGINT, catch_signal)
+    signal.signal(signal.SIGINT, catch_signal)
     
-if __name__=='__main__':
+    url=load_config()
     
-    start()
+    start(url)