/Python/Checking number of connected users hotspot ubuntu

Checking number of connected users hotspot ubuntu

#!/usr/bin/python3.6   
import subprocess
import re
import requests

# Store Mac address of all nodes here
saved = {
    'xx:xx:xx:xx:xx:xx': 'My laptop',
}

# Set wireless interface using ifconfig
interface = "wlp4s0"

mac_regex = re.compile(r'([a-zA-Z0-9]{2}:){5}[a-zA-Z0-9]{2}')


def parse_arp():
    arp_out = subprocess.check_output(f'arp -e -i {interface}', shell=True).decode('utf-8')
    if 'no match found' in arp_out:
        return None

    result = []
    for lines in arp_out.strip().split('\n'):
        line = lines.split()
        if interface in line and '(incomplete)' not in line:
            for element in line:
                # If its a mac addr
                if mac_regex.match(element):
                    result.append((line[0], element))
    return result


def get_mac_vendor(devices):
    num = 0
    for device in devices:
        try:
            url = f"http://api.macvendors.com/{device[1]}"
            try:
                vendor = requests.get(url).text
            except Exception as e:
                print(e)
                vendor = None

        except Exception as e:
            print("Error occured while getting mac vendor", e)

        num += 1
        print_device(device, num, vendor)

def print_device(device, num=0, vendor=None):
    device_name = saved[device[1]] if device[1] in saved else 'unrecognised !!'

    print(f'\n{num})', device_name,  '\nVendor:', vendor, '\nMac:', device[1], '\nIP: ',device[0])

if __name__ == '__main__':
    print('Retrieving connected devices ..')

    devices = parse_arp()

    if not devices:
        print('No devices found!')

    else:
        print('Retrieving mac vendors ..')
        try:
            get_mac_vendor(devices)

        except KeyboardInterrupt as e:
            num = 0
            for device in devices:
                num += 1
                print_device(device, num)

Source: askubuntu.com

Similar codes

0 vote223 views1 answer
Python pid control
Python
2 votes213 views2 answers
If else in dictionary comprehension python
Python
1 vote205 views1 answer
How to create routes in flask
Python
0 vote203 views1 answer
How to get foregine kew field in models.py
Python
2 votes227 views1 answer
Python keep value recursive function
Python
0 vote204 views1 answer
How to open a widget using sputil.get
Python
0 vote210 views1 answer
Print less than specific number in one row python
Python
4 votes209 views4 answers
Set python3.7 as default ubuntu
Python
7 votes261 views2 answers
No module named 'PyQt5.QtWebEngineWidgets'
Python
0 vote201 views1 answer
Np.linspace() in python
Python