Added webmasters money transfers

Author: trezorg

README.md

@@ -354,6 +354,26 @@ API Items
     res = client.PaymentsManage.delete(71)
 
 
+### Money transfers ###
+
+
+###### List of money transfers ######
+
+    res = client.MoneyTransfers.get()
+    res = client.MoneyTransfers.get(limit=2)
+    res = client.MoneyTransfers.getOne(2)
+    res = client.MoneyTransfers.get(sender='sender')
+    res = client.MoneyTransfers.get(currency='USD')
+
+###### Manage money transfers ######
+
+    res = client.MoneyTransfersManage.create(
+        currency='USD',
+        recipient='recipient',
+        comment='comment',
+        sum=10)
+
+
 Notes
 ------
 

pyadmitad/items/__init__.py

@@ -9,3 +9,4 @@
 from pyadmitad.items.products import *
 from pyadmitad.items.announcements import *
 from pyadmitad.items.payments import *
+from pyadmitad.items.money_transfers import *

pyadmitad/items/base.py

@@ -50,6 +50,22 @@ def sanitize_integer_value(value, name, blank=False):
                 return value
         raise ValueError("Invalid integer value '%s': %s" % (name, value))
 
+    @staticmethod
+    def sanitize_float_value(value, name, blank=False):
+        if not value:
+            if not blank:
+                raise ValueError("Blank float value '%s': %s" % (name, value))
+            return value
+        if type(value) in (float, int):
+            return str(value)
+        elif type(value) == str:
+            try:
+                float(value)
+                return value
+            except ValueError:
+                raise ValueError("Invalid float value '%s': %s" % (name, value))
+        raise ValueError("Invalid float value '%s': %s" % (name, value))
+
     @staticmethod
     def sanitize_integer_array(values, name, blank=False):
         if not values:

pyadmitad/items/money_transfers.py

@@ -0,0 +1,88 @@
+from copy import deepcopy
+from pyadmitad.items.base import Item
+
+
+__all__ = (
+    'MoneyTransfers',
+    'MoneyTransfersManage',
+)
+
+
+class MoneyTransfersBase(Item):
+
+    ORDERING = ('date_created',)
+    FILTERING = {
+        'sender': Item.to_unicode,
+        'recipient': Item.to_unicode,
+        'currency': Item.to_unicode,
+    }
+
+
+class MoneyTransfers(Item):
+    """
+    List of webmaster money transfers
+
+    Required scope - "webmaster_money_transfers"
+    """
+    URL = Item.prepare_url('webmaster_money_transfers')
+    SINGLE_URL = Item.prepare_url('webmaster_money_transfer/%(id)s')
+
+    def get(self, **kwargs):
+        """
+        res = client.MoneyTransfers.get()
+        res = client.MoneyTransfers.get(limit=2)
+
+        """
+        kwargs['url'] = self.URL
+        return self.transport.set_method('GET').\
+            set_pagination(**kwargs).set_filtering(**kwargs).\
+            set_ordering(**kwargs).request(**kwargs)
+
+    def getOne(self, _id, **kwargs):
+        """
+        res = client.MoneyTransfers.getOne(_id=2)
+        res = client.MoneyTransfers.getOne(2)
+        """
+        kwargs['url'] = self.SINGLE_URL
+        kwargs['id'] = self.sanitize_id(_id)
+        return self.transport.set_method('GET').request(**kwargs)
+
+
+class MoneyTransfersManage(Item):
+    """
+    Manage webmaster money transfers
+
+    Required scope - "manage_webmaster_money_transfers"
+    """
+    CREATE_FIELDS = {
+        'comment': lambda x: Item.sanitize_string_value(x, 'comment'),
+        'recipient': lambda x: Item.sanitize_string_value(x, 'recipient'),
+        'currency': lambda x: Item.sanitize_currency(x, 'currency'),
+        'sum': lambda x: Item.sanitize_float_value(x, 'sum')
+    }
+
+    CREATE_URL = Item.prepare_url('webmaster_money_transfer/create')
+
+    @staticmethod
+    def sanitize_fields(fields, **kwargs):
+        data = deepcopy(kwargs)
+        for field in fields:
+            data[field] = fields[field](data.get(field))
+        return dict([(key, value) for (key, value) in data.items() if value])
+
+    def create(self, **kwargs):
+        """
+        Create a webmaster money transfers
+
+        res = client.MoneyTransfersManage.create(
+            sender='webmaster',
+            recipient='recipient',
+            sum=200,
+            currency='USD',
+            comment='comment')
+
+        """
+        data = self.sanitize_fields(self.CREATE_FIELDS, **kwargs)
+        kwargs['url'] = self.CREATE_URL
+        return self.transport.set_method('POST').\
+            set_data(data).request(**kwargs)

pyadmitad/items/websites.py

@@ -98,7 +98,8 @@ class WebsitesManage(Item):
             x, 'atnd_hits', blank=True)
     }
 
-    def sanitize_fields(self, fields, **kwargs):
+    @staticmethod
+    def sanitize_fields(fields, **kwargs):
         data = deepcopy(kwargs)
         for field in fields:
             data[field] = fields[field](data.get(field))

pyadmitad/tests/test_money_transfer.py

@@ -0,0 +1,112 @@
+# -*- coding: utf-8 -*-
+
+import unittest
+from pyadmitad.items import MoneyTransfers, MoneyTransfersManage
+from pyadmitad.tests.base import BaseTestCase
+
+
+MONEY_TRANSFER_CREATE_DATA = dict(
+    currency='USD',
+    comment="test",
+    recipient="admitadppvweb",
+    sum='200.12',
+)
+
+
+class MoneyTransfersTestCase(BaseTestCase):
+
+    def test_get_money_transfers_request(self):
+        self.set_mocker(MoneyTransfers.URL, limit=1)
+        result = {
+            "_meta": {
+                "count": 6,
+                "limit": 1,
+                "offset": 0
+            },
+            "results": [
+                {
+                    "comment": "test",
+                    "sender": {
+                        "username": "webmaster1",
+                        "id": 96
+                    },
+                    "sum": 200.0,
+                    "currency": "USD",
+                    "date_created": "2013-12-06T12:28:29",
+                    "recipient": {
+                        "username": "admitadppvweb",
+                        "id": 100
+                    },
+                    "id": 8
+                }
+            ]
+        }
+        self.mocker.result(result)
+        self.mocker.replay()
+        res = self.client.MoneyTransfers.get(limit=1)
+        self.assertIn(u'results', res)
+        self.assertIn(u'_meta', res)
+        self.assertIsInstance(res[u'results'], list)
+        self.assertIsInstance(res[u'_meta'], dict)
+        self.assertEqual(res[u'_meta'][u'limit'], 1)
+        self.assertEqual(res[u'results'][0][u'currency'], u'USD')
+        self.mocker.verify()
+
+    def test_get_money_transfers_request_with_id(self):
+        self.set_mocker(MoneyTransfers.SINGLE_URL, id=8, with_pagination=False)
+        result = {
+            "comment": "test",
+            "sender": {
+                "username": "webmaster1",
+                "id": 96
+            },
+            "sum": 200.0,
+            "currency": "USD",
+            "date_created": "2013-12-06T12:28:29",
+            "recipient": {
+                "username": "admitadppvweb",
+                "id": 100
+            },
+            "id": 8
+        }
+        self.mocker.result(result)
+        self.mocker.replay()
+        res = self.client.MoneyTransfers.getOne(8)
+        self.assertEqual(res[u'id'], 8)
+        self.mocker.verify()
+
+
+class MoneyTransfersManageTestCase(BaseTestCase):
+
+    def test_create_payments_request(self):
+        self.set_mocker(MoneyTransfersManage.CREATE_URL,
+                        method='POST',
+                        with_pagination=False,
+                        data=MONEY_TRANSFER_CREATE_DATA)
+        result = {
+            "comment": "test",
+            "sender": {
+                "username": "webmaster1",
+                "id": 96
+            },
+            "sum": 200.12,
+            "currency": "USD",
+            "date_created": "2013-12-06T12:28:29",
+            "recipient": {
+                "username": "admitadppvweb",
+                "id": 100
+            },
+            "id": 9
+        }
+        self.mocker.result(result)
+        self.mocker.replay()
+        res = self.client.MoneyTransfersManage.create(
+            **MONEY_TRANSFER_CREATE_DATA)
+        self.assertEqual(res[u'comment'], u'test')
+        self.assertEqual(res[u'currency'], u'USD')
+        self.assertEqual(res[u'sum'], 200.12)
+        self.assertEqual(res[u'sender'][u'username'], u'webmaster1')
+        self.mocker.verify()
+
+if __name__ == '__main__':
+    unittest.main()

pyadmitad/transport.py

@@ -443,9 +443,13 @@ def request(self, **kwargs):
             raise AttributeError(
                 'Absent url parameter. Use set_url method or pass '
                 'url parameter in this method.')
-        response = self.api_request(
-            self._url, method=self._method,
-            headers=self._headers, data=self._data, debug=self._debug)
+        requests_kwargs = {
+            'method': self._method,
+            'headers': self._headers,
+            'data': self._data,
+            'debug': self._debug
+        }
+        response = self.api_request(self._url, **requests_kwargs)
         return kwargs.get('handler', self._handle_response)(response)
 
     def __call__(self, **kwargs):