using requests.Session for better optimization

Author: James Lin

rest_framework_queryset/__init__.py

@@ -2,6 +2,6 @@
 from __future__ import unicode_literals
 from __future__ import absolute_import
 
-__version__ = '0.1.2'
+__version__ = '0.2.2'
 
 from .queryset import RestFrameworkQuerySet

rest_framework_queryset/queryset.py

@@ -8,7 +8,8 @@
 
 class BaseAPIQuerySet(object):
     def __init__(self, url, *args, **kwargs):
-        self.request_method = requests.get
+        _req_session = requests.Session()
+        self.request_method = _req_session.get
         self.url = url
         self.args = args
         self.kwargs = kwargs

tests.py

@@ -31,7 +31,7 @@ def test_filter_chain(self):
 
     def test_filter_call(self):
         fake_response = MagicMock(json=lambda:{'count': 10, 'results': range(10)})
-        with patch('rest_framework_queryset.queryset.requests.get', return_value=fake_response) as mock_get:
+        with patch('rest_framework_queryset.queryset.requests.Session.get', return_value=fake_response) as mock_get:
             qs = RestFrameworkQuerySet('/api/')
             qs1 = qs.filter(a=123)
             self.assertEqual(list(qs1), list(range(10)))
@@ -42,7 +42,7 @@ def test_filter_call(self):
 
     def test_get_call(self):
         fake_response = MagicMock(json=lambda:{'count': 10, 'results': range(10)})
-        with patch('rest_framework_queryset.queryset.requests.get', return_value=fake_response) as mock_get:
+        with patch('rest_framework_queryset.queryset.requests.Session.get', return_value=fake_response) as mock_get:
             qs = RestFrameworkQuerySet('/api/')
             with self.assertRaises(MultipleObjectsReturned):
                 qs1 = qs.get(a=123)
@@ -51,13 +51,13 @@ def test_get_call(self):
 
     def test_count_call(self):
         fake_response = MagicMock(json=lambda:{'count': 10, 'results': range(10)})
-        with patch('rest_framework_queryset.queryset.requests.get', return_value=fake_response) as mock_get:
+        with patch('rest_framework_queryset.queryset.requests.Session.get', return_value=fake_response) as mock_get:
             count = RestFrameworkQuerySet('/api/').count()
             self.assertEqual(count, 10)
 
     def test_all(self):
         fake_response = MagicMock(json=lambda:{'count': 10, 'results': range(10)})
-        with patch('rest_framework_queryset.queryset.requests.get', return_value=fake_response) as mock_get:
+        with patch('rest_framework_queryset.queryset.requests.Session.get', return_value=fake_response) as mock_get:
             qs = RestFrameworkQuerySet('/api/').all()
             self.assertEqual(list(qs), list(range(10)))