/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

1 vote132 views1 answer
How to use get-pip.py
Python
0 vote141 views1 answer
Fix the debug_mode = false django
Python
0 vote131 views1 answer
Can you use the astro a50 with a phone
Python
27 votes116 views2 answers
Time py
Python
0 vote120 views2 answers
Pytest parametrize
Python
1 vote126 views2 answers
Find full name regular expression
Python
0 vote130 views1 answer
Los mejores script de phython
Python
16 votes180 views6 answers
Fillna with mean pandas
Python
0 vote135 views1 answer
Python slack webhook
Python
0 vote129 views1 answer
Python cv2 convert image to binary
Python