#!/usr/bin/env python3
import random
import socket
import sys

from os import path

import pyrad.packet

from pyrad.client import Client
from pyrad.dictionary import Dictionary


def send_accounting_packet(srv, req):
    try:
        srv.SendPacket(req)
    except pyrad.client.Timeout:
        print('RADIUS server does not reply')
        sys.exit(1)
    except socket.error as error:
        print('Network error: ' + error[1])
        sys.exit(1)


def main(path_to_dictionary):
    srv = Client(server='127.0.0.1',
                 secret=b'Kah3choteereethiejeimaeziecumi',
                 dict=Dictionary(path_to_dictionary))

    req = srv.CreateAcctPacket(**{
        'User-Name': 'wichert',
        'NAS-IP-Address': '192.168.1.10',
        'NAS-Port': 0,
        'NAS-Identifier': 'trillian',
        'Called-Station-Id': '00-04-5F-00-0F-D1',
        'Calling-Station-Id': '00-01-24-80-B3-9C',
        'Framed-IP-Address': '10.0.0.100',
    })

    print('Sending accounting start packet')
    req['Acct-Status-Type'] = 'Start'
    send_accounting_packet(srv, req)

    print('Sending accounting stop packet')
    req['Acct-Status-Type'] = 'Stop'
    req['Acct-Input-Octets'] = random.randrange(2**10, 2**30)
    req['Acct-Output-Octets'] = random.randrange(2**10, 2**30)
    req['Acct-Session-Time'] = random.randrange(120, 3600)
    req['Acct-Terminate-Cause'] = random.choice(['User-Request', 'Idle-Timeout'])
    send_accounting_packet(srv, req)


if __name__ == '__main__':
    dictionary = path.join(path.dirname(path.abspath(__file__)), 'dictionary')
    main(dictionary)