/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 vote100 views3 answers
Unsplash source
Python
2 votes100 views1 answer
Raise RuntimeError("populate() isn't reentrant")
Python
1 vote112 views1 answer
How to say hello with name in python
Python
13 votes133 views6 answers
Tensorflow check gpu
Python
2 votes114 views1 answer
Django template tag to display current year
Python
1 vote188 views1 answer
Flask clear session
Python Flask
0 vote106 views1 answer
Find all unique substring permutations of a string of a specific length python
Python
0 vote104 views1 answer
Find word position in string python
Python
97 votes115 views6 answers
How to read a csv file in python
Python
1 vote109 views1 answer
Python dont exit script after print
Python