Wrap functions that might fail some time, so it will retry to execute that function n times
Project description
Installation
pip3 install trython
About
The main use for this package is when making requests with the requests module. Sometimes the network might be unavailable, Or the server (the endpiont) might be temporarily down, We don't want our script to quit just because of that (although we can put everything in a loop and catch error's, We don't want to do that because that will still lose the progress of the previous loop, and is a bit messy).
This is where trython comes in, it gives you two very approachable techniques to solve this.
From here on I'll be using requests.get
as the function we want to add error handling to.
Approach 1
Replacing requests.get
with an alternitave error handled one.
import requests
import trython
requests.get = trython.wrap(requests.get)
Now when you call requests.get
and it throws an error, we will catch that and try to run the request again - without you even knowing.
Approach 2
Decorating your own function.
import requests
import trython
@trython.wrap()
def requests_get(url):
return requests.get(url)
Now when you call requests_get
the same logic will be applied.
Approach 3
Creating a temporary function with a context manager.
import requests
import trython
with trython.context_wrap(requests.get, validator=validators.requests_json_validator, time_to_sleep=1) as get:
response = get('https://jsonplaceholder.typicode.com/posts').json()
with trython.context_wrap(requests.get, validator=validators.requests_xml_validator, time_to_sleep=1) as get:
response = get('https://www.cs.utexas.edu/~mitra/csFall2015/cs329/lectures/xml/xslplanes.1.xml.txt').text
This is very usefull if you want to have diffrent validators for the same function, as shown above.
Options
The wrap function takes in a couple of arguments:
func
: (callable) This is the function that we are adding error handling to.number_of_attempts
: (int) default=5 The number of times you want to retry before finally raising the error.time_to_sleep
: (int) default=30 Time to sleep between retries.errors_to_catch
: (tuple) default=(Exception, ) Which errors you want to handle.validator
: (callable) default=None A function that will validate if the return of the function is valid, if the function throws an error or returns false - it will be treated as if thefunc
thew an error.on_exception_callback
: (callable) default=None A function to be called every time there's an exception raised byfunc
, use this if you would like to not retry some errors - based on the error message.on_raise_callback
: (callable) default=None A function to be called when we finaly raise the error, this function should take the exception as an argument.
More about validtators
Let's take another look at validators, because without them this whole thing is useless. A very common reason that a script will quit is when you are hitting a api endpiont, and you expect a certain data type, and you go ahead and call some function and that data that will give you an error when it doesn't understand that data it got, without validators there's no way for us to know that.
That's when we create a validator - when we expect the result of a given function to be something, but might a times be something else entirely. We create a function that takes in the func's return value and we check if it satisfies us, And if the validator function returns false or throws an error, then we retry, otherwise - we know we got the right data.
I have included some predefined validators which you can import like this: from trython import validators
.
Feel free to make pr with some other helpful validators.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
File details
Details for the file trython-3.3.0-py3-none-any.whl
.
File metadata
- Download URL: trython-3.3.0-py3-none-any.whl
- Upload date:
- Size: 6.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.10.0 pkginfo/1.7.0 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.60.0 CPython/3.8.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c09cc70c3bb1debd607b47e840f295b5d97741072594f41ed348266c9a7b6b6a |
|
MD5 | 3e7b88c0007cdb2755cc6aa9a1f0d65a |
|
BLAKE2b-256 | 907718fb93775fbfb0e1d279fa8921c102097e14589aff6b3c9445f6fe941c52 |