diff --git a/toriptables2.py b/toriptables2.py index c310d40..dea5d8e 100755 --- a/toriptables2.py +++ b/toriptables2.py @@ -1,5 +1,4 @@ #! /usr/bin/env python -#! /usr/bin/env python # Written by Rupe version 2 # """ @@ -17,6 +16,7 @@ from sys import stdout, stderr from atexit import register from argparse import ArgumentParser from json import load +from pynotify import init, Notification from urllib2 import urlopen, URLError @@ -61,12 +61,14 @@ DNSPort %s try: my_public_ip = load(urlopen('http://jsonip.com'))['ip'] except URLError: - my_public_ip = None - my_public_ip = getoutput('wget -qO - v4.ifconfig.co') - if not my_public_ip: - exit(" \033[91m[!]\033[0m Can't get public ip address!") - print(" {0}".format( - "[\033[92m+\033[0m] Your IP is \033[92m%s\033[0m" % my_public_ip)) + print(" \033[91m[!]\033[0m Can't get public ip address!") + else: + print(" {0}".format( + "[\033[92m+\033[0m] Your IP is \033[92m%s\033[0m" % my_public_ip)) + init('toriptables2') + my_ip = Notification("Your public IP address is %s" % my_public_ip) + my_ip.show() + except CalledProcessError as err: print("[!] Command failed: %s" % err.cmd) @@ -120,6 +122,14 @@ if __name__ == '__main__': load_tables.flush_iptables_rules() print(" {0}".format( "[\033[93m!\033[0m] Anonymizer status \033[91m[OFF]\033[0m")) + try: + my_real_ip = load(urlopen('http://jsonip.com'))['ip'] + except URLError: + pass + else: + init('toriptables2') + my_own_ip = Notification("Your public IP address is %s" % my_real_ip) + my_own_ip.show() else: parser.print_help() except Exception as err: