108 lines
2.8 KiB
Python
108 lines
2.8 KiB
Python
|
#! /usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# Copyright (C) 2011 Sebastian Pipping <sebastian@pipping.org>
|
||
|
|
# Licensed under GPL v3 or later
|
||
|
|
|
||
|
|
from __future__ import print_function
|
||
|
|
import urllib
|
||
|
|
import urllib2
|
||
|
|
import simplejson as json
|
||
|
|
from decimal import Decimal
|
||
|
|
import os
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
import sys
|
||
|
|
sys.path.append('.')
|
||
|
|
from freitagslib.item import Item
|
||
|
|
from freitagslib.tools import require_type
|
||
|
|
|
||
|
|
|
||
|
|
DEPOSIT_CASH = 1
|
||
|
|
DEPOSIT_BANK = 2
|
||
|
|
|
||
|
|
|
||
|
|
_BASE_URL = 'http://devcat.someserver.de:13805/api2/'
|
||
|
|
|
||
|
|
|
||
|
|
def _make_auth_opener(username, password):
|
||
|
|
password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
|
||
|
|
password_mgr.add_password(None, _BASE_URL, username, password)
|
||
|
|
handler = urllib2.HTTPBasicAuthHandler(password_mgr)
|
||
|
|
return urllib2.build_opener(handler)
|
||
|
|
|
||
|
|
_auth_opener = _make_auth_opener(os.environ['BARCODE_PLUGIN_USER'], os.environ['BARCODE_PLUGIN_PASS'])
|
||
|
|
|
||
|
|
|
||
|
|
def _request(api_rel_url, para_map, method):
|
||
|
|
url = _BASE_URL + api_rel_url
|
||
|
|
data = urllib.urlencode(para_map)
|
||
|
|
|
||
|
|
if method == 'GET':
|
||
|
|
url = '%s?%s' % (url, data)
|
||
|
|
req = urllib2.Request(url)
|
||
|
|
elif method == 'POST':
|
||
|
|
req = urllib2.Request(url, data)
|
||
|
|
else:
|
||
|
|
raise ValueError('Unsupported method "%s"' % method)
|
||
|
|
|
||
|
|
response = _auth_opener.open(req)
|
||
|
|
return response.read()
|
||
|
|
|
||
|
|
|
||
|
|
def get_user_name_from_auth_blob(authblob):
|
||
|
|
content = _request('auth/user/', {'authblob':authblob}, method='GET')
|
||
|
|
d = json.loads(content)
|
||
|
|
return d['username']
|
||
|
|
|
||
|
|
|
||
|
|
def buy_item(item_id, user_name):
|
||
|
|
require_type(int, item_id)
|
||
|
|
|
||
|
|
content = _request('buyable/item/' + urllib.quote(str(item_id)),
|
||
|
|
{'user':user_name, 'deposit':0, 'amount':1}, method='POST')
|
||
|
|
if content != 'Created':
|
||
|
|
raise ValueError('Server says "%s"' % content)
|
||
|
|
|
||
|
|
|
||
|
|
def get_item(item_id):
|
||
|
|
require_type(int, item_id)
|
||
|
|
|
||
|
|
content = _request('buyable/item/' + urllib.quote(str(item_id)),
|
||
|
|
{}, method='GET')
|
||
|
|
d = json.loads(content)
|
||
|
|
return Item(d)
|
||
|
|
|
||
|
|
|
||
|
|
def get_balance(user_name):
|
||
|
|
content = _request('account/balance/',
|
||
|
|
{'user':user_name}, method='GET')
|
||
|
|
d = json.loads(content)
|
||
|
|
return Decimal(d['balance'])
|
||
|
|
|
||
|
|
|
||
|
|
def deposit(amount, transaction_type, user_name):
|
||
|
|
require_type(Decimal, amount)
|
||
|
|
require_type(int, transaction_type)
|
||
|
|
|
||
|
|
content = _request('account/transactions/transact/',
|
||
|
|
{'user':user_name, 'amount':amount, 'type':transaction_type}, method='POST')
|
||
|
|
if content != 'OK':
|
||
|
|
raise ValueError('Server says "%s"' % content)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
user_name = get_user_name_from_auth_blob(os.environ['BARCODE_PLUGIN_TEST_AUTH_BLOB'])
|
||
|
|
print('User name: ' + user_name)
|
||
|
|
|
||
|
|
buy_item(1, user_name)
|
||
|
|
|
||
|
|
balance = get_balance(user_name)
|
||
|
|
print('Balance is %f, type is "%s"' % (balance, type(balance).__name__))
|
||
|
|
|
||
|
|
deposit(Decimal('0.01'), DEPOSIT_CASH, user_name)
|
||
|
|
deposit(Decimal('0.01'), DEPOSIT_BANK, user_name)
|
||
|
|
|
||
|
|
item = get_item(1)
|
||
|
|
print('Item "%s" is %f Euro each' % (item.name, item.price))
|