You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
5 years ago
|
import requests
|
||
|
|
from torrequest import TorRequest
|
||
|
|
from fake_useragent import UserAgent
|
||
|
|
|
||
|
|
|
||
|
|
class Client:
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.session = requests.Session()
|
||
|
|
self.session.headers['UserAgent'] = UserAgent().random
|
||
|
|
|
||
|
|
def request(self, *args, method='GET', **kwargs):
|
||
|
|
if method == 'GET':
|
||
|
|
return self.get(*args, **kwargs)
|
||
|
|
elif method == 'POST':
|
||
|
|
return self.post(*args, **kwargs)
|
||
|
|
elif method == 'PUT':
|
||
|
|
return self.put(*args, **kwargs)
|
||
|
|
elif method == 'PATCH':
|
||
|
|
return self.patch(*args, **kwargs)
|
||
|
|
elif method == 'DELETE':
|
||
|
|
return self.delete(*args, **kwargs)
|
||
|
|
else:
|
||
|
|
raise Exception('Invalid HTTP method specified!')
|
||
|
|
|
||
|
|
def get(self, *args, **kwargs):
|
||
|
|
return self.session.get(*args, **kwargs)
|
||
|
|
|
||
|
|
def post(self, *args, **kwargs):
|
||
|
|
return self.session.get(*args, **kwargs)
|
||
|
|
|
||
|
|
def put(self, *args, **kwargs):
|
||
|
|
return self.session.put(*args, **kwargs)
|
||
|
|
|
||
|
|
def patch(self, *args, **kwargs):
|
||
|
|
return self.session.patch(*args, **kwargs)
|
||
|
|
|
||
|
|
def delete(self,*args, **kwargs):
|
||
|
|
return self.session.delete(*args, **kwargs)
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
self.session.close()
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.session.headers['UserAgent'] = UserAgent().random
|
||
|
|
self.session.cookies.clear()
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def __exit__(self, *args):
|
||
|
|
self.close()
|
||
|
|
|
||
|
|
|
||
|
|
class TorClient(Client):
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__()
|
||
|
|
self.tr = TorRequest(*args, **kwargs)
|
||
|
|
self.tr.session.headers['UserAgent'] = UserAgent().random
|
||
|
|
self.session = self.tr
|
||
|
|
|
||
|
|
def new_identity(self):
|
||
|
|
self.tr.reset_identity()
|
||
|
|
print("[+] Changed IP to %s: " % self.get('https://ident.me').text)
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.new_identity()
|
||
|
|
self.tr.session.cookies.clear()
|
||
|
|
self.tr.session.headers['UserAgent'] = UserAgent().random
|