client-barcode: Add basic networking code

This commit is contained in:
Sebastian Pipping 2011-10-05 11:35:56 +02:00
parent b2438f2981
commit 7a1ee41ca2
5 changed files with 136 additions and 0 deletions

View File

@ -5,6 +5,7 @@
# Licensed under GPL v3 or later
from __future__ import print_function
import freitagslib.network as net
import sys
import decimal

View File

View File

@ -0,0 +1,14 @@
#! /usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2011 Sebastian Pipping <sebastian@pipping.org>
# Licensed under GPL v3 or later
from decimal import Decimal
class Item:
def __init__(self, d):
self.deposit = Decimal(d['deposit'])
self.id = int(d['id'])
self.name = d['name']
self.price = Decimal(d['price'])

View File

@ -0,0 +1,107 @@
#! /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))

View File

@ -0,0 +1,14 @@
#! /usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2011 Sebastian Pipping <sebastian@pipping.org>
# Licensed under GPL v3 or later
def _raise_type_error(value, wanted, got):
raise TypeError('Value <%s> is of type "%s" but type "%s" is required' \
% (str(value), wanted.__name__, got.__name__))
def require_type(wanted, value):
got = type(value)
if not got is wanted:
_raise_type_error(value, wanted, got)