1 files changed, 53 insertions, 0 deletions
diff --git a/machammer/network.py b/machammer/network.py
new file mode 100644
index 0000000..17608ce
--- /dev/null
+++ b/machammer/network.py
@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+"""Network-related machammer functions."""
+
+from .functions import call, check_output
+from .system_profiler import SystemProfile
+
+
+def get_ports(type='Ethernet'):
+    """Return all devices of type (Ethernet, AirPort)"""
+    profile = SystemProfile('Network')
+    return [x for x in profile if x['type'] == type]
+
+
+def set_wifi_power(on=True):
+    """Set AirPort power to on (True) or off (False)"""
+    state = 'on' if on else 'off'
+    for i in get_ports('AirPort'):
+        call('/usr/sbin/networksetup', '-setairportpower', i['interface'], state)
+
+
+def get_wifi_power():
+    """Get AirPort power state"""
+    results = []
+    for i in get_ports('AirPort'):
+        iface = i['interface']
+        r = check_output('/usr/sbin/networksetup', '-getairportpower', iface)
+        results.append(r.split(': ')[1])
+
+    return 'On' in results
+
+
+def is_wired(primary=False):
+    """Do we have an "active" Ethernet connection?"""
+    for p in get_ports('Ethernet'):
+        r = check_output('/sbin/ifconfig', p['interface'])
+        active = 'status: active' in r
+        return get_primary(p['interface']) if primary else active
+
+
+def is_wireless():
+    return not is_wired()
+
+
+def get_primary(port=None):
+    """Return device node of primary network interface"""
+    try:
+        route = check_output('/sbin/route', 'get', 'default').split('\n')
+    except Exception as e:
+        raise Exception('Failed to get default route: %s' % e)
+
+    for i in [x for x in route if 'interface: ' in x]:
+        p = i.split(': ')[1]
+        return p == port if port else p