FilesExpand file tree

 remove_old_python3_support

/

utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (30 loc) · 919 Bytes

 remove_old_python3_support

/

utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (30 loc) · 919 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

import os

from sys import getsizeof

from requests import HTTPError

def ensure_dir(path):

if not os.path.isdir(path):

os.makedirs(path)

def compute_memory_usage(obj):

seen = set()

stack = [obj]

returned = 0

while stack:

obj = stack.pop()

if id(obj) in seen:

continue

seen.add(id(obj))

returned += getsizeof(obj)

if isinstance(obj, dict):

for key, value in obj.items():

stack.append(key)

stack.append(value)

elif isinstance(obj, (list, tuple, set, frozenset)):

stack.extend(obj)

return returned

def raise_for_status(resp):

try:

resp.raise_for_status()

except HTTPError as e:

raise HTTPError(

'{0.request.method} {0.request.url}: {0.status_code}\n\n{0.content}'.format(e.response),

response=resp, request=resp.request)