joeni

PingPing module for Joeni. Provides and implements the functionality required to

interact with PingPing.

"""

import requests

from .models import user

from django.core.cache import caches

def balance(user_id):

    """ Returns the current balance associated with the given account.

    The returned value may be a cached value, so if you top up your balance and

    check immediately, be advised it may not be the correct result, but it will

    be updated.

    Future idea: If the balance is below a certain treshold (for example €2), I

    can hardwire the function so that it ignores the cache. Most people will not

    top up their balance until it's below a certain point.

    While logging the times the result is higher than the cached result is

    handy, caches are deleted after a certain time, so the info is lost. I'll

    see what I can do in the future. Possibly collect all the times an account

    is topped up, and take the median of those results to see what is the most

    likely amount that somebody will top up their balance.

    """

    # Check cache first

    cache = cashes['pingping']

    cached = cache.get(user_id)

    if cached is not None:

        return cached

    else:

        result = update_balance(user_id)

        # Updating cache

        cache.set(user_id, result, timeout=3600)  # Cached for 1 hour

        cache.close()

        return result

def update_balance(user_id):

    """ Connects to the PingPing website and looks for the current budget. """

    user_data = user.objects.get(id=user_id)

    username = user_data.name

    password = user_data.password

    link = "https://uhasselt-pxl.mynetpay.be/Account/Login"

    # Get CSRF protection token

    first_call = requests.get(link)

    text = first_call.text

    begin = text.find('__RequestVerificationToken')

    begin = text.find('value="', begin)

    end = text.find('" ', begin)

    token = text[begin+len('value="'):end]

    cookies = first_call.cookies

    response = requests.post(link, data={

        'Username':username,

        'LoginType':'Student',

        'Password':password,

        '__RequestVerificationToken':token,

        }, cookies=cookies)

    html_response = response.text

    start = html_response.find(": € ")

    offset = len(": € ")

    return  html_response[start+offset:start+offset+4]