/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

3 votes229 views5 answers
Pandas split column into multiple columns
Python
0 vote227 views1 answer
How to get index of closest value in list python
Python
0 vote240 views1 answer
How to find determinant in numpy
Python
1 vote264 views1 answer
Python towers of hanoi recursive
Python
1 vote234 views1 answer
Python how to inspect pyd for functions
Python
0 vote269 views1 answer
Python launch ipython from script
Python
0 vote235 views1 answer
Successful=true for number in range (3) print ("Attempt") if successful: print ("Successful") breal
Python
1 vote210 views1 answer
Fastest sort python
Python
2 votes247 views1 answer
Interviewbit with Python questions solutions
Python
23 votes203 views6 answers
How to check if a number is even or odd in python
Python