freevpn.py

import re
import requests
from bs4 import BeautifulSoup


__author__ = "wrathdev GGSIPUResultTracker orekihotaro demonnezuko varmaincorp hotarooreki nezukodemon"

RE_PASS = re.compile(r'(<li><strong>Password:<\/strong>)(?P<password>[^<]+)(<\/li>)')

TEXT_ONLINE = 'Online'
VPNS = ('me', 'eu', 'se', 'im', 'it', 'be')


def get_password(username):
    """
    Get the password for the vpn server.

    Args:
        username : Name of the freevpn server
                   Example:
                        freevpn.me
                        freevpn.se
    """
    passw = None
    html  = requests.get('https://'+ username + '/accounts').text

    match = RE_PASS.search(html)
    if match:
        passw = match.groups()[1]
    
    return passw.strip() if passw else None

def get_vpn_status():
    """
    Get the status of FreeVPN Servers
    It scrapes the FreeVPN website for server status and details.

    Return:
        list of dict with data
        Example dict:-
            {'name':'freevpn.me', 'loc':'France', 'online':true, 'load':'9'}
    """
    html = requests.get('https://freevpn.me/').text
    soup = BeautifulSoup(html, 'lxml')

    table = soup.find_all(
        'table', attrs={'class': "table table-striped table-bordered dataTable"})
    if table:
        table = table[0]
        tbody = table.tbody
        trows = tbody.find_all('tr')
        for row in trows:
            name = None
            loc = None
            online = False
            load = None
            tcols = row.find_all('td')
            for i, tcol in enumerate(tcols):
                if i == 1:
                    name = tcol.text.strip().lower()
                elif i == 2:
                    loc = tcol.text.strip()
                elif i == 4:
                    if tcol.text.strip() == TEXT_ONLINE:
                        online = True
                elif i == 8:
                    load = tcol.text.strip()[:-1]
            if name:
                yield {'name': name, 'loc': loc,
                       'online': online, 'load': load}