Commit 86a6e92

author

MF Softworks

authored

[script] added args parsing for easy customization

1 parent 9ed59bf commit 86a6e92Copy full SHA for 86a6e92

File tree

1 file changed

+28

-16

lines changed

monitor.py

1 file changed

+28

-16

lines changed

`‎monitor.py‎`

Lines changed: 28 additions & 16 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,16 @@`
`1`		`-import socket, time, json, datetime, platform, psutil, requests, pprint, uuid`
	`1`	`+import argparse, socket, time, json, datetime, platform, psutil, requests, pprint, uuid`
	`2`	`+`
	`3`	`+# parse args`
	`4`	`+parser = argparse.ArgumentParser(description='Monitoring script to send system info to a tracking server')`
	`5`	`+parser.add_argument('-d', '--dest', default='http://localhost:8080/', help='API Endpoint for Monitoring Data (Defaults to http://localhost:8080/)')`
	`6`	`+parser.add_argument('-i', '--interval', default=5, type=int, help='Interval between checks (Seconds. Defaults to 5 seconds)')`
	`7`	`+parser.add_argument('-a', '--attempts', default=30, type=int, help='Attempts to send data when sending failes (Defaults to 30)')`
	`8`	`+parser.add_argument('-t', '--timeout', default=60, type=int, help='Timeout between resend attempts (Seconds. Defaults to 60. If attempts is reached script will die)')`
	`9`	`+args = parser.parse_args()`
	`10`	`+`
	`11`	`+# Factor in sleep for bandwidth checking`
	`12`	`+if args.interval >= 2:`
	`13`	`+ args.interval -= 2`
`2`	`14`
`3`	`15`	`def main():`
`4`	`16`	`# Hostname Info`
`@@ -25,11 +37,11 @@ def main():`
`25`	`37`	`# Try fixes issues with connected 'disk' such as CD-ROMS, Phones, etc.`
`26`	`38`	`try:`
`27`	`39`	`disk = {`
`28`		`- "name" : x.device,`
`29`		`- "mount_point" : x.mountpoint,`
`30`		`- "type" : x.fstype,`
`31`		`- "total_size" : psutil.disk_usage(x.mountpoint).total,`
`32`		`- "used_size" : psutil.disk_usage(x.mountpoint).used,`
	`40`	`+ "name" : x.device,`
	`41`	`+ "mount_point" : x.mountpoint,`
	`42`	`+ "type" : x.fstype,`
	`43`	`+ "total_size" : psutil.disk_usage(x.mountpoint).total,`
	`44`	`+ "used_size" : psutil.disk_usage(x.mountpoint).used,`
`33`	`45`	`"percent_used" : psutil.disk_usage(x.mountpoint).percent`
`34`	`46`	`}`
`35`	`47`
`@@ -39,7 +51,7 @@ def main():`
`39`	`51`	`except:`
`40`	`52`	`print("")`
`41`	`53`
`42`		`- # Bandwidth Info`
	`54`	`+ # Bandwidth Info`
`43`	`55`	`network_stats = get_bandwidth()`
`44`	`56`	`print("Network:\n\tTraffic in:",network_stats["traffic_in"] / 1e+6,"\n\tTraffic out:",network_stats["traffic_out"] / 1e+6)`
`45`	`57`
`@@ -66,7 +78,7 @@ def main():`
`66`	`78`	`nic["address6"] = snic.address`
`67`	`79`	`nics.append(nic)`
`68`	`80`	`print("\tNIC:",nic["name"], "\tMAC:", nic["mac"], "\tIPv4 Address:",nic["address"], "\tIPv4 Subnet:", nic["netmask"], "\tIPv6 Address:", nic["address6"])`
`69`		`-`
	`81`	`+`
`70`	`82`	`# Platform Info`
`71`	`83`	`system = {`
`72`	`84`	`"name" : platform.system(),`
`@@ -81,7 +93,7 @@ def main():`
`81`	`93`
`82`	`94`	`# System UUID`
`83`	`95`	`sys_uuid = uuid.getnode()`
`84`		`-`
	`96`	`+`
`85`	`97`	`# Set Machine Info`
`86`	`98`	`machine = {`
`87`	`99`	`"hostname" : hostname,`
`@@ -122,21 +134,21 @@ def get_bandwidth():`
`122`	`134`	`current_in = 0`
`123`	`135`	`else:`
`124`	`136`	`current_in = net2_in - net1_in`
`125`		`-`
	`137`	`+`
`126`	`138`	`if net1_out > net2_out:`
`127`	`139`	`current_out = 0`
`128`	`140`	`else:`
`129`	`141`	`current_out = net2_out - net1_out`
`130`		`-`
	`142`	`+`
`131`	`143`	`network = {"traffic_in" : current_in, "traffic_out" : current_out}`
`132`	`144`	`return network`
`133`	`145`
`134`	`146`	`def send_data(data):`
`135`	`147`	`# Attempt to send data up to 30 times`
`136`		`- for attempt in range(30):`
	`148`	`+ for attempt in range(args.attempts):`
`137`	`149`	`try:`
`138`	`150`	`# endpoint = monitoring server`
`139`		`- endpoint = "http://monitor.localhost.local/api/"`
	`151`	`+ endpoint = args.dest`
`140`	`152`	`response = requests.post(url = endpoint, data = data)`
`141`	`153`	`print("\nPOST:")`
`142`	`154`	`print("Response:", response.status_code)`
`@@ -153,12 +165,12 @@ def send_data(data):`
`153`	`165`	`except requests.exceptions.RequestException as e:`
`154`	`166`	`print("\nPOST Error:\n",e)`
`155`	`167`	`# Sleep 1 minute before retrying`
`156`		`- time.sleep(60)`
	`168`	`+ time.sleep(args.timeout)`
`157`	`169`	`else:`
`158`		`- # If no connection established for half an hour, kill script`
	`170`	`+ # If no connection established for attempts*timeout, kill script`
`159`	`171`	`exit(0)`
`160`	`172`
`161`	`173`	`while True:`
`162`	`174`	`main()`
`163`	`175`	`print("-----------------------------------------------------------------")`
`164`		`- time.sleep(3)`
	`176`	`+ time.sleep(args.interval)`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 86a6e92

File tree

1 file changed

1 file changed

`‎monitor.py‎`

0 commit comments