def disconnect_clicked(self, e):
select = self.ap_list.get_selection()
profile = self.get_profile()
NetCTL.stop(profile)
self.notifications.show_notification("Stopping profile.", "Stopping profile: {}".format(profile))
InterfaceControl.down(self.interface_name)
self.start_scan(None)
def disconnect_clicked(self, e):
select = self.ap_list.get_selection()
profile = self.get_profile()
NetCTL.stop(profile)
self.notifications.show_notification(
"Stopping profile.", "Stopping profile: {}".format(profile))
InterfaceControl.down(self.interface_name)
self.start_scan(None)
def disconnect_clicked(self, e):
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
profile = "netgui_" + network_ssid
interface_name = get_interface()
NetCTL.stop(self, profile)
InterfaceControl.down(self, interface_name)
self.start_scan(None)
def connect_clicked(self, e):
self.disable_buttons()
profile_name = self.get_profile()
if profile_name is not None:
self.notifications.show_notification(
"Profile found!",
"Found profile {} for this network. Connecting " +
"using pre-existing profile!".format(profile_name))
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
self.statusbar.push(self.context,
"Connecting to {}".format(profile_name))
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
self.statusbar.push(self.context,
"Connected to {}".format(profile_name))
else:
if self.NoWifiMode == False:
self.notifications.show_notification(
"No profile found.",
"There is no profile for this network. Creating one now!")
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
profile = "netgui_" + network_ssid
if Path(profile_dir, profile).is_file():
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
else:
network_security = self.get_security(select)
if network_security == "Open":
key = "none"
else:
key = self.get_network_password()
self.generate_config.create_wireless_config(
network_ssid, self.interface_name, network_security,
key)
try:
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
except Exception as e:
pass
elif self.NoWifiMode == 1:
self.notifications.show_notification(
"We are in no-wifi mode.",
"We are in no-wifi mode, connecting to existing profile.")
select = self.ap_list.get_selection()
nwm_profile = self.get_profile()
try:
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(nwm_profile)
except:
pass
self.start_scan(None)
GObject.timeout_add_seconds(5, self.enable_buttons)
def connect_clicked(self, e):
self.disable_buttons()
profile_name = self.get_profile()
if profile_name is not None:
self.notifications.show_notification("Profile found!", "Found profile {} for this network. Connecting " +
"using pre-existing profile!".format(profile_name))
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
self.statusbar.push(self.context, "Connecting to {}".format(profile_name))
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
self.statusbar.push(self.context, "Connected to {}".format(profile_name))
else:
if self.NoWifiMode == False:
self.notifications.show_notification("No profile found.", "There is no profile for this network. Creating one now!")
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
profile = "netgui_" + network_ssid
if Path(profile_dir, profile).is_file():
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
else:
network_security = self.get_security(select)
if network_security == "Open":
key = "none"
else:
key = self.get_network_password()
self.generate_config.create_wireless_config(network_ssid, self.interface_name, network_security, key)
try:
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(profile_name)
except Exception as e:
pass
elif self.NoWifiMode == 1:
self.notifications.show_notification("We are in no-wifi mode.", "We are in no-wifi mode, connecting to existing profile.")
select = self.ap_list.get_selection()
nwm_profile = self.get_profile()
try:
InterfaceControl.down(self.interface_name)
NetCTL.stop_all()
NetCTL.start(nwm_profile)
except:
pass
self.start_scan(None)
GObject.timeout_add_seconds(5, self.enable_buttons)
def disconnect_all_clicked(self, e):
self.notifications.show_notification("Stopping all profiles.", "Stopping all profiles!")
NetCTL.stop_all()
InterfaceControl.down(self.interface_name)
self.start_scan(None)
def disconnect_all_clicked(self, e):
self.notifications.show_notification("Stopping all profiles.",
"Stopping all profiles!")
NetCTL.stop_all()
InterfaceControl.down(self.interface_name)
self.start_scan(None)
def disconnect_all(self, e):
interface_name = get_interface()
NetCTL.stop_all(None)
InterfaceControl.down(self, interface_name)
self.start_scan(None)
def connect_clicked(self, does_profile_exist, profile_name):
# process a connection request from the user
if does_profile_exist is 1:
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
#n = Notify.Notification.new("Found existing profile.",
# "NetCTL found an existing profile for this network. Connecting to " +
# network_ssid + " using profile " + profile_name)
#n.show()
net_interface = self.interface_name
self.interface_control.down(net_interface)
NetCTL.stop_all()
NetCTL.start(profile_name)
#n = Notify.Notification.new("Connected to new network!", "You are now connected to " + network_ssid,
# "dialog-information")
#n.show()
elif does_profile_exist == 0:
if self.NoWifiMode == 0:
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
print("nSSID = " + network_ssid)
profile = "netgui_" + network_ssid
print("profile = " + profile)
net_interface = self.interface_name
if Path(profile_dir, profile).is_file():
self.interface_control.down(net_interface)
NetCTL.stop_all(self)
NetCTL.start(profile)
# n = Notify.Notification.new("Connected to new network!", "You are now connected to " +
# network_ssid, "dialog-information")
# n.show()
else:
network_security = self.get_security(select)
if network_security == "Open":
key = "none"
else:
key = self.get_network_pw()
self.generate_config.create_wireless_config(network_ssid, self.interface_name, network_security, key)
try:
InterfaceControl.down(self, net_interface)
NetCTL.stop_all(self)
NetCTL.start(self, profile)
# n = Notify.Notification.new("Connected to new network!", "You are now connected to " +
# network_ssid, "dialog-information")
# n.show()
except Exception as e:
pass
# n = Notify.Notification.new("Error!", "There was an error. The error was: " + str(e) +
# ". Please report an issue at the github page if it persists.",
# "dialog-information")
# n.show()
# Notify.uninit()
elif self.NoWifiMode == 1:
select = self.ap_list.get_selection()
nwm_profile = self.get_ssid(select)
net_interface = get_interface()
try:
InterfaceControl.down(self, net_interface)
NetCTL.stop_all(self)
NetCTL.start(self, nwm_profile)
# n = Notify.Notification.new("Connected to new profile!", "You are now connected to " + nwm_profile,
# "dialog-information")
# n.show()
except:
pass
# n = Notify.Notification.new("Error!", "There was an error. Please report an issue at the "
# + "github page if it persists.", "dialog-information")
# n.show()
# Notify.uninit()
self.start_scan(self)
class NetGUI(Gtk.Window):
def __init__(self):
# Init Vars
self.scanning = False
self.APindex = 0
self.p = None
self.builder = Gtk.Builder()
GObject.type_register(GtkSource.View)
self.builder.add_from_file(str(Path(str(program_loc), "UI.glade")))
self.dialog = self.builder.get_object("passwordDialog")
self.ap_list = self.builder.get_object("treeview1")
self.ap_store = Gtk.ListStore(str, str, str, str)
self.interface_name = ""
self.NoWifiMode = 0
self.interface_control = InterfaceControl()
self.generate_config = GenConfig(profile_dir)
self.init_ui()
def init_ui(self):
is_connected()
# Create a "Builder", which basically allows me to import the Glade file for a complete interface.
# I love Glade, btw. So much quicker than manually coding everything.
# Grab the "window1" attribute from UI.glade, and set it to show everything.
window = self.builder.get_object("mainWindow")
window.connect("delete-event", Gtk.main_quit)
# Get the OnScan button in case we are going to run in NoWifiMode
scan_button = self.builder.get_object("scanAPsTool")
# Setup the main area of netgui: The network list.
self.ap_list.set_model(self.ap_store)
# Set Up Columns
# renderer1 = The Cell renderer. Basically allows for text to show.
# column1 = The actual setup of the column. Arguments = title, CellRenderer, textIndex)
# Actually append the column to the treeview.
ssid_cell_renderer = Gtk.CellRendererText()
ssid_column = Gtk.TreeViewColumn("SSID", ssid_cell_renderer, text=0)
ssid_column.set_resizable(True)
ssid_column.set_expand(True)
self.ap_list.append_column(ssid_column)
connect_quality_cell_renderer = Gtk.CellRendererText()
connect_quality_column = Gtk.TreeViewColumn("Connection Quality", connect_quality_cell_renderer, text=1)
connect_quality_column.set_resizable(True)
connect_quality_column.set_expand(True)
self.ap_list.append_column(connect_quality_column)
security_type_cell_renderer = Gtk.CellRendererText()
security_type_column = Gtk.TreeViewColumn("Security Type", security_type_cell_renderer, text=2)
security_type_column.set_resizable(True)
security_type_column.set_expand(True)
self.ap_list.append_column(security_type_column)
connected_cell_renderer = Gtk.CellRendererText()
connected_column = Gtk.TreeViewColumn("Connected?", connected_cell_renderer, text=3)
connected_column.set_resizable(True)
connected_column.set_expand(True)
self.ap_list.append_column(connected_column)
# Set TreeView as Re-orderable
self.ap_list.set_reorderable(True)
# Set all the handlers I defined in glade to local functions.
handlers = {
"onSwitch": self.on_switch,
"onExit": self.on_btn_exit,
"onAboutClicked": self.about_clicked,
"onScan": self.start_scan,
"onConnect": self.profile_exists,
"onDConnect": self.disconnect_clicked,
"onPrefClicked": self.preferences_clicked,
"onHelpClicked": self.on_help_clicked,
"onIssueReport": self.report_issue,
"onDAll": self.disconnect_all,
"onEditorActivate": self.open_editor
}
# Connect all the above handlers to actually call the functions.
self.builder.connect_signals(handlers)
# Hardcode (relative) image paths
APScanToolImg = self.builder.get_object("image1")
APScanToolImg.set_from_file(str(Path(img_loc, "APScan.png")))
ConnectToolImg = self.builder.get_object("image2")
ConnectToolImg.set_from_file(str(Path(img_loc, "connect.png")))
dConnectToolImg = self.builder.get_object("image3")
dConnectToolImg.set_from_file(str(Path(img_loc, "disconnect.png")))
prefToolImg = self.builder.get_object("image5")
prefToolImg.set_from_file(str(Path(img_loc, "preferences.png")))
exitToolImg = self.builder.get_object("image4")
exitToolImg.set_from_file(str(Path(img_loc, "exit.png")))
# Populate profiles menu
profile_menu = self.builder.get_object("profilesMenu")
profile_menu.set_submenu(Gtk.Menu())
profiles = os.listdir("/etc/netctl/")
# Iterate through profiles directory, and add to "Profiles" Menu #
for i in profiles:
if Path("/etc/netctl/" + i).is_file():
profile_menu.get_submenu().append(Gtk.MenuItem(label=i))
#This should automatically detect their wireless device name. I'm not 100% sure
#if it works on every computer, but we can only know from multiple tests. If
#it doesn't work, I will re-implement the old way.
self.interface_name = get_interface()
if self.interface_name == "":
#n = Notify.Notification.new("Could not detect interface!", "No interface was detected. Now running in " +
# "No-Wifi Mode. Scan Button is disabled.", "dialog-information")
#n.show()
self.no_wifi_scan()
self.NoWifiMode = 1
scan_button.props.sensitive = False
print(str(self.NoWifiMode))
elif args.nowifi:
self.no_wifi_scan()
self.NoWifiMode = 1
scan_button.props.sensitive = False
else:
#self.startScan(None)
self.NoWifiMode = 0
# Start initial scan
#Notify.init("NetGUI")
window.show_all()
def open_editor(self, e):
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
if network_ssid is None:
profile_edit_window = NetGUIProfileEditor(None)
profile_edit_window.show()
else:
profile = str(Path("/", "etc", "netctl", "netgui_" + network_ssid))
profile_edit_window = NetGUIProfileEditor(profile)
profile_edit_window.show()
def no_wifi_scan(self):
aps = {}
profiles = os.listdir(Path("/", "etc", "netctl"))
i = 0
self.NoWifiMode = 1
global args
args.nowifi = True
for profile in profiles:
if Path("/", "etc", "netctl" + profile).is_file():
aps["row" + str(i)] = self.ap_store.append([profile, "", "", ""])
self.ap_store.set(aps["row" + str(i)], 1, "N/A in No-Wifi mode.")
self.ap_store.set(aps["row" + str(i)], 2, "N/A.")
if is_connected() is False:
self.ap_store.set(aps["row" + str(i)], 3, "No")
else:
connected_network = check_output(self, "netctl list | sed -n 's/^\* //p'").strip()
if profile in connected_network:
self.ap_store.set(aps["row" + str(i)], 3, "Yes")
else:
self.ap_store.set(aps["row" + str(i)], 3, "No")
i += 1
def start_scan(self, e):
run_scan = ScanRoutines(self.interface_name, scan_file, status_dir)
run_scan.scan()
self.check_scan()
def check_scan(self):
sf = open(scan_file, 'r')
real_dir = sf.readline()
real_dir = real_dir.strip()
sf.close()
print(real_dir)
shutil.move(real_dir, Path(status_dir, "final_results.log"))
with open(Path(status_dir, "final_results.log")) as tsv:
self.ap_store.clear()
r = csv.reader(tsv, dialect='excel-tab')
aps = {}
i = 0
for row in r:
network = row[2]
print(ascii(network))
if r"\x00" in network:
continue
elif network is "":
continue
else:
aps["row" + str(i)] = self.ap_store.append([network, "", "", ""])
quality = row[0]
if int(quality) <= -100:
percent = "0%"
elif int(quality) >= -50:
percent = "100%"
else:
final_quality = (2 * (int(quality) + 100))
percent = str(final_quality) + "%"
if network == "":
pass
else:
self.ap_store.set(aps["row" + str(i)], 1, percent)
security = row[1]
if "WPA" in security:
encryption = "WPA"
elif "OPENSSID" in security:
encryption = "Open"
elif "WPS" in security:
encryption = "WPS"
elif "WEP" in security:
encryption = "WEP"
else:
encryption = "Open"
if network == "":
pass
else:
self.ap_store.set(aps["row" + str(i)], 2, encryption)
if is_connected() is False:
if network == "":
pass
else:
if network == "":
pass
else:
self.ap_store.set(aps["row" + str(i)], 3, "No")
else:
connected_network = check_output(self, "netctl list | sed -n 's/^\* //p'").strip()
if network in connected_network:
if network == "":
pass
else:
self.ap_store.set(aps["row" + str(i)], 3, "Yes")
else:
if network == "":
pass
else:
self.ap_store.set(aps["row" + str(i)], 3, "No")
i += 1
def on_switch(self, e):
self.ap_store.clear()
self.no_wifi_scan()
self.NoWifiMode = 1
global args
args.nowifi = True
def on_btn_exit(self, e):
if self.p is None:
pass
else:
self.p.terminate()
Gtk.main_quit()
sys.exit()
def about_clicked(self, e):
# Getting the about dialog from UI.glade
about_dialog = self.builder.get_object("aboutDialog")
# Opening the about dialog.
about_dialog.run()
# Hiding the about dialog. Read in "prefDialog" for why we hide, not destroy.
about_dialog.hide()
def profile_exists(self, e):
skip_no_prof_conn = 0
found_profile = 0
select = self.ap_list.get_selection() # Get selected network
ssid = self.get_ssid(select) # Get SSID of selected network.
for profile in os.listdir(Path("/", "etc", "netctl")):
if Path("/", "etc", "netctl", profile).is_file(): # Is it a file, not dir?
with open(Path("/", "etc", "netctl", profile), 'r') as current_profile:
current_profile_name = profile
for line in current_profile:
if "ESSID" in line.strip():
essid_name = line[6:]
if str(ssid).lower() in essid_name.lower():
skip_no_prof_conn = 1
if found_profile is 1:
break
else:
self.connect_clicked(1, current_profile_name)
found_profile = 1
else:
pass
else:
pass
else:
pass
if skip_no_prof_conn is 1:
pass
else:
self.connect_clicked(0, None)
def connect_clicked(self, does_profile_exist, profile_name):
# process a connection request from the user
if does_profile_exist is 1:
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
#n = Notify.Notification.new("Found existing profile.",
# "NetCTL found an existing profile for this network. Connecting to " +
# network_ssid + " using profile " + profile_name)
#n.show()
net_interface = self.interface_name
self.interface_control.down(net_interface)
NetCTL.stop_all()
NetCTL.start(profile_name)
#n = Notify.Notification.new("Connected to new network!", "You are now connected to " + network_ssid,
# "dialog-information")
#n.show()
elif does_profile_exist == 0:
if self.NoWifiMode == 0:
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
print("nSSID = " + network_ssid)
profile = "netgui_" + network_ssid
print("profile = " + profile)
net_interface = self.interface_name
if Path(profile_dir, profile).is_file():
self.interface_control.down(net_interface)
NetCTL.stop_all(self)
NetCTL.start(profile)
# n = Notify.Notification.new("Connected to new network!", "You are now connected to " +
# network_ssid, "dialog-information")
# n.show()
else:
network_security = self.get_security(select)
if network_security == "Open":
key = "none"
else:
key = self.get_network_pw()
self.generate_config.create_wireless_config(network_ssid, self.interface_name, network_security, key)
try:
InterfaceControl.down(self, net_interface)
NetCTL.stop_all(self)
NetCTL.start(self, profile)
# n = Notify.Notification.new("Connected to new network!", "You are now connected to " +
# network_ssid, "dialog-information")
# n.show()
except Exception as e:
pass
# n = Notify.Notification.new("Error!", "There was an error. The error was: " + str(e) +
# ". Please report an issue at the github page if it persists.",
# "dialog-information")
# n.show()
# Notify.uninit()
elif self.NoWifiMode == 1:
select = self.ap_list.get_selection()
nwm_profile = self.get_ssid(select)
net_interface = get_interface()
try:
InterfaceControl.down(self, net_interface)
NetCTL.stop_all(self)
NetCTL.start(self, nwm_profile)
# n = Notify.Notification.new("Connected to new profile!", "You are now connected to " + nwm_profile,
# "dialog-information")
# n.show()
except:
pass
# n = Notify.Notification.new("Error!", "There was an error. Please report an issue at the "
# + "github page if it persists.", "dialog-information")
# n.show()
# Notify.uninit()
self.start_scan(self)
def get_network_pw(self):
ret = self.dialog.run()
self.dialog.hide()
entry = self.builder.get_object("userEntry")
if ret == 1:
password = entry.get_text()
return password
def get_ssid(self, selection):
model, treeiter = selection.get_selected()
if treeiter is not None:
return model[treeiter][0]
@staticmethod
def get_security(selection):
model, treeiter = selection.get_selected()
if treeiter is not None:
security_type = model[treeiter][2]
security_type = security_type.lower()
return security_type
def disconnect_clicked(self, e):
select = self.ap_list.get_selection()
network_ssid = self.get_ssid(select)
profile = "netgui_" + network_ssid
interface_name = get_interface()
NetCTL.stop(self, profile)
InterfaceControl.down(self, interface_name)
self.start_scan(None)
# n = Notify.Notification.new("Disconnected from network!", "You are now disconnected from " + network_ssid,
# "dialog-information")
# n.show()
def disconnect_all(self, e):
interface_name = get_interface()
NetCTL.stop_all(None)
InterfaceControl.down(self, interface_name)
self.start_scan(None)
# n = Notify.Notification.new("Disconnected from all networks!", "You are now disconnected from all networks.",
# "dialog-information")
# n.show()
def preferences_clicked(self, e):
Preferences(program_loc)
#TODO: Write help file!
def on_help_clicked(self, e):
pass
def report_issue(self, e):
pass