Skip to content
This repository was archived by the owner on Nov 20, 2022. It is now read-only.

Commit f1023b3

Browse files
author
Anthony Jiang
authored
Merge pull request #1 from tmxkn1/develop
Merge changes into latest upstream master
2 parents bc0c250 + ce5b6c3 commit f1023b3

2 files changed

Lines changed: 6 additions & 19 deletions

File tree

src/pythonpancakes/api.py

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2,39 +2,29 @@
22
import re
33
import requests
44

5-
from requests.adapters import HTTPAdapter
6-
from requests.packages.urllib3.util.retry import Retry
7-
85

96
class PancakeSwapAPI:
107
__BASE_URL = "https://api.pancakeswap.info/api/v2/"
8+
119
"""
1210
Basic API request wrapper for PancakeSwap
1311
Scott Burlovich (github.com/scottburlovich)
14-
Last Update: May 9, 2021
12+
Last Update: Jul 28, 2021
1513
"""
1614

1715
def __init__(self, base_url=__BASE_URL):
1816
self.base_url = base_url
1917
self.request_timeout = 60
20-
self.session = requests.session()
21-
retries = Retry(total=3, backoff_factor=0.5, status_forcelist=[502, 503, 504])
22-
self.session.mount('https://', HTTPAdapter(max_retries=retries))
23-
24-
def __enter__(self):
25-
return self
26-
27-
def __exit__(self):
28-
self.session.close()
2918

3019
def __get(self, request_url: str):
3120
"""
3221
GET request wrapper
3322
:param request_url: str
3423
"""
35-
response = self.session.get(request_url, timeout=self.request_timeout)
36-
response.raise_for_status()
37-
return json.loads(response.content.decode('utf-8'))
24+
with requests.Session() as session:
25+
response = session.get(request_url, timeout=self.request_timeout)
26+
response.raise_for_status()
27+
return json.loads(response.content.decode('utf-8'))
3828

3929
def summary(self):
4030
"""

tests/test_api_unittest.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,6 @@ class TestAPI(unittest.TestCase):
66
def setUp(self):
77
self.ps = PancakeSwapAPI()
88

9-
def tearDown(self):
10-
self.ps.__exit__()
11-
129
def test_summary(self):
1310
"""Verify that valid data is returned from the summary() request"""
1411
response = self.ps.summary()

0 commit comments

Comments
 (0)