Compare commits
24 Commits
Author | SHA1 | Date | |
---|---|---|---|
cb567bf5c0 | |||
bb6d1be1b5 | |||
c546c8f5eb | |||
9acc2d2e65 | |||
51a15874f2 | |||
af8f764396 | |||
65b555f2bd | |||
319b4414df | |||
b71cb0c263 | |||
0936dee06c | |||
a1c6fb06a0 | |||
c57bf99976 | |||
995b45464f | |||
ebe28c948a | |||
34d7230faf | |||
3235991cce | |||
3fba1d6b3d | |||
88c8d5a9a0 | |||
382749a8d8 | |||
0c877e315b | |||
c95f642fee | |||
da0276f9a6 | |||
f89f91e969 | |||
56c2ca1b6e |
@ -4,7 +4,7 @@
|
||||
|
||||
__author__ = """Eriks Karls"""
|
||||
__email__ = 'eriks@72.lv'
|
||||
__version__ = '0.23.4.9'
|
||||
__version__ = '0.24.0.1'
|
||||
|
||||
from erepublik import classes, constants, utils
|
||||
from erepublik.citizen import Citizen
|
||||
|
@ -13,43 +13,43 @@ __all__ = ['SlowRequests', 'CitizenAPI']
|
||||
class SlowRequests(Session):
|
||||
last_time: datetime.datetime
|
||||
timeout: datetime.timedelta = datetime.timedelta(milliseconds=500)
|
||||
uas: List[str] = [
|
||||
_uas: List[str] = [
|
||||
# Chrome
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.183 Safari/537.36',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.125 Safari/537.36',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.106 Safari/537.36',
|
||||
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36',
|
||||
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.183 Safari/537.36',
|
||||
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.125 Safari/537.36',
|
||||
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36',
|
||||
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.106 Safari/537.36',
|
||||
|
||||
# FireFox
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:84.0) Gecko/20100101 Firefox/84.0',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:83.0) Gecko/20100101 Firefox/83.0',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:82.0) Gecko/20100101 Firefox/82.0',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:81.0) Gecko/20100101 Firefox/81.0',
|
||||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:80.0) Gecko/20100101 Firefox/80.0',
|
||||
'Mozilla/5.0 (X11; Linux x86_64; rv:84.0) Gecko/20100101 Firefox/84.0',
|
||||
'Mozilla/5.0 (X11; Linux x86_64; rv:83.0) Gecko/20100101 Firefox/83.0',
|
||||
'Mozilla/5.0 (X11; Linux x86_64; rv:82.0) Gecko/20100101 Firefox/82.0',
|
||||
'Mozilla/5.0 (X11; Linux x86_64; rv:81.0) Gecko/20100101 Firefox/81.0',
|
||||
'Mozilla/5.0 (X11; Linux x86_64; rv:80.0) Gecko/20100101 Firefox/80.0',
|
||||
]
|
||||
debug: bool = False
|
||||
|
||||
def __init__(self, proxies: Dict[str, str] = None):
|
||||
def __init__(self, proxies: Dict[str, str] = None, user_agent: str = None):
|
||||
super().__init__()
|
||||
if proxies:
|
||||
self.proxies = proxies
|
||||
if user_agent is None:
|
||||
user_agent = random.choice(self._uas)
|
||||
self.request_log_name = utils.get_file(utils.now().strftime("debug/requests_%Y-%m-%d.log"))
|
||||
self.last_time = utils.now()
|
||||
self.headers.update({
|
||||
'User-Agent': random.choice(self.uas)
|
||||
})
|
||||
self.headers.update({'User-Agent': user_agent})
|
||||
|
||||
@property
|
||||
def as_dict(self):
|
||||
return dict(last_time=self.last_time, timeout=self.timeout, user_agent=self.headers['User-Agent'],
|
||||
request_log_name=self.request_log_name, debug=self.debug)
|
||||
return dict(last_time=self.last_time, timeout=self.timeout, cookies=self.cookies.get_dict(), debug=self.debug,
|
||||
user_agent=self.headers['User-Agent'], request_log_name=self.request_log_name, proxies=self.proxies)
|
||||
|
||||
def request(self, method, url, *args, **kwargs):
|
||||
self._slow_down_requests()
|
||||
@ -73,13 +73,13 @@ class SlowRequests(Session):
|
||||
args.update({'kwargs': kwargs})
|
||||
|
||||
if data:
|
||||
args.update({"data": data})
|
||||
args.update({'data': data})
|
||||
|
||||
if json:
|
||||
args.update({"json": json})
|
||||
args.update({'json': json})
|
||||
|
||||
if params:
|
||||
args.update({"params": params})
|
||||
args.update({'params': params})
|
||||
|
||||
body = f"[{utils.now().strftime('%F %T')}]\tURL: '{url}'\tMETHOD: {method}\tARGS: {args}\n"
|
||||
with open(self.request_log_name, 'ab') as file:
|
||||
@ -91,13 +91,13 @@ class SlowRequests(Session):
|
||||
if self.debug:
|
||||
if resp.history and not redirect:
|
||||
for hist_resp in resp.history:
|
||||
self._log_request(hist_resp.request.url, "REDIRECT")
|
||||
self._log_request(hist_resp.request.url, 'REDIRECT')
|
||||
self._log_response(hist_resp.request.url, hist_resp, redirect=True)
|
||||
|
||||
fd_path = 'debug/requests'
|
||||
fd_time = self.last_time.strftime('%Y/%m/%d/%H-%M-%S')
|
||||
fd_name = utils.slugify(url[len(Citizen.url):])
|
||||
fd_extra = "_REDIRECT" if redirect else ""
|
||||
fd_extra = '_REDIRECT' if redirect else ""
|
||||
|
||||
try:
|
||||
utils.json.loads(resp.text)
|
||||
@ -120,6 +120,10 @@ class CitizenBaseAPI:
|
||||
self._req = SlowRequests()
|
||||
self.token = ""
|
||||
|
||||
@property
|
||||
def as_dict(self):
|
||||
return dict(url=self.url, request=self._req.as_dict, token=self.token)
|
||||
|
||||
def post(self, url: str, data=None, json=None, **kwargs) -> Response:
|
||||
return self._req.post(url, data, json, **kwargs)
|
||||
|
||||
@ -140,7 +144,7 @@ class CitizenBaseAPI:
|
||||
|
||||
class ErepublikAnniversaryAPI(CitizenBaseAPI):
|
||||
def _post_main_collect_anniversary_reward(self) -> Response:
|
||||
return self.post(f"{self.url}/main/collect-anniversary-reward", data={"_token": self.token})
|
||||
return self.post(f"{self.url}/main/collect-anniversary-reward", data={'_token': self.token})
|
||||
|
||||
# 12th anniversary endpoints
|
||||
def _get_anniversary_quest_data(self) -> Response:
|
||||
@ -151,7 +155,7 @@ class ErepublikAnniversaryAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/main/map-rewards-unlock", data=data)
|
||||
|
||||
def _post_map_rewards_speedup(self, node_id: int, currency_amount: int) -> Response:
|
||||
data = {'nodeId': node_id, '_token': self.token, "currencyCost": currency_amount}
|
||||
data = {'nodeId': node_id, '_token': self.token, 'currencyCost': currency_amount}
|
||||
return self.post(f"{self.url}/main/map-rewards-speedup", data=data)
|
||||
|
||||
def _post_map_rewards_claim(self, node_id: int, extra: bool = False) -> Response:
|
||||
@ -161,7 +165,7 @@ class ErepublikAnniversaryAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/main/map-rewards-claim", data=data)
|
||||
|
||||
def _post_main_wheel_of_fortune_spin(self, cost) -> Response:
|
||||
return self.post(f"{self.url}/main/wheeloffortune-spin", data={'_token': self.token, "_currentCost": cost})
|
||||
return self.post(f"{self.url}/main/wheeloffortune-spin", data={'_token': self.token, '_currentCost': cost})
|
||||
|
||||
def _post_main_wheel_of_fortune_build(self) -> Response:
|
||||
return self.post(f"{self.url}/main/wheeloffortune-build", data={'_token': self.token})
|
||||
@ -183,7 +187,7 @@ class ErepublikArticleAPI(CitizenBaseAPI):
|
||||
def _post_main_article_comments_create(self, message: str, article_id: int, parent: int = 0) -> Response:
|
||||
data = dict(_token=self.token, message=message, articleId=article_id)
|
||||
if parent:
|
||||
data.update({"parentId": parent})
|
||||
data.update({'parentId': parent})
|
||||
return self.post(f"{self.url}/main/articleComments/create", data=data)
|
||||
|
||||
def _post_main_donate_article(self, article_id: int, amount: int) -> Response:
|
||||
@ -202,13 +206,13 @@ class ErepublikArticleAPI(CitizenBaseAPI):
|
||||
|
||||
class ErepublikCompanyAPI(CitizenBaseAPI):
|
||||
def _post_economy_assign_to_holding(self, factory_id: int, holding_id: int) -> Response:
|
||||
data = dict(_token=self.token, factoryId=factory_id, action="assign", holdingCompanyId=holding_id)
|
||||
data = dict(_token=self.token, factoryId=factory_id, action='assign', holdingCompanyId=holding_id)
|
||||
return self.post(f"{self.url}/economy/assign-to-holding", data=data)
|
||||
|
||||
def _post_economy_create_company(self, industry_id: int, building_type: int = 1) -> Response:
|
||||
data = {"_token": self.token, "company[industry_id]": industry_id, "company[building_type]": building_type}
|
||||
data = {'_token': self.token, "company[industry_id]": industry_id, "company[building_type]": building_type}
|
||||
return self.post(f"{self.url}/economy/create-company", data=data,
|
||||
headers={"Referer": f"{self.url}/economy/create-company"})
|
||||
headers={'Referer': f"{self.url}/economy/create-company"})
|
||||
|
||||
def _get_economy_inventory_items(self) -> Response:
|
||||
return self.get(f"{self.url}/economy/inventory-items/")
|
||||
@ -229,12 +233,12 @@ class ErepublikCompanyAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/economy/train", data=data)
|
||||
|
||||
def _post_economy_upgrade_company(self, factory: int, level: int, pin: str = None) -> Response:
|
||||
data = dict(_token=self.token, type="upgrade", companyId=factory, level=level, pin="" if pin is None else pin)
|
||||
data = dict(_token=self.token, type='upgrade', companyId=factory, level=level, pin="" if pin is None else pin)
|
||||
return self.post(f"{self.url}/economy/upgrade-company", data=data)
|
||||
|
||||
def _post_economy_work(self, action_type: str, wam: List[int] = None, employ: Dict[int, int] = None) -> Response:
|
||||
data: Dict[str, Union[int, str]] = dict(action_type=action_type, _token=self.token)
|
||||
if action_type == "production":
|
||||
if action_type == 'production':
|
||||
if employ is None:
|
||||
employ = {}
|
||||
if wam is None:
|
||||
@ -257,7 +261,7 @@ class ErepublikCompanyAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/economy/work", data=data)
|
||||
|
||||
def _post_economy_work_overtime(self) -> Response:
|
||||
data = dict(action_type="workOvertime", _token=self.token)
|
||||
data = dict(action_type='workOvertime', _token=self.token)
|
||||
return self.post(f"{self.url}/economy/workOvertime", data=data)
|
||||
|
||||
def _post_economy_job_market_apply(self, citizen_id: int, salary: float) -> Response:
|
||||
@ -267,16 +271,16 @@ class ErepublikCompanyAPI(CitizenBaseAPI):
|
||||
def _post_economy_resign(self) -> Response:
|
||||
return self.post(f"{self.url}/economy/resign",
|
||||
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
||||
data={"_token": self.token, "action_type": "resign"})
|
||||
data={'_token': self.token, 'action_type': 'resign'})
|
||||
|
||||
def _post_economy_sell_company(self, factory_id: int, pin: int = None, sell: bool = True) -> Response:
|
||||
data = dict(_token=self.token, pin="" if pin is None else pin)
|
||||
if sell:
|
||||
data.update({"sell": "sell"})
|
||||
data.update({'sell': 'sell'})
|
||||
else:
|
||||
data.update({"dissolve": factory_id})
|
||||
data.update({'dissolve': factory_id})
|
||||
return self.post(f"{self.url}/economy/sell-company/{factory_id}",
|
||||
data=data, headers={"Referer": self.url})
|
||||
data=data, headers={'Referer': self.url})
|
||||
|
||||
|
||||
class ErepublikCountryAPI(CitizenBaseAPI):
|
||||
@ -287,7 +291,7 @@ class ErepublikCountryAPI(CitizenBaseAPI):
|
||||
quality: int = None) -> Response:
|
||||
data = dict(countryId=country_id, action=action, _token=self.token, value=value, quality=quality)
|
||||
return self.post(f"{self.url}/main/country-donate", data=data,
|
||||
headers={"Referer": f"{self.url}/country/economy/Latvia"})
|
||||
headers={'Referer': f"{self.url}/country/economy/Latvia"})
|
||||
|
||||
|
||||
class ErepublikEconomyAPI(CitizenBaseAPI):
|
||||
@ -309,20 +313,20 @@ class ErepublikEconomyAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/economy/activateBooster", data=data)
|
||||
|
||||
def _post_economy_activate_house(self, quality: int) -> Response:
|
||||
data = dict(action="activate", quality=quality, type="house", _token=self.token)
|
||||
data = dict(action='activate', quality=quality, type='house', _token=self.token)
|
||||
return self.post(f"{self.url}/economy/activateHouse", data=data)
|
||||
|
||||
def _post_economy_donate_items_action(self, citizen_id: int, amount: int, industry: int,
|
||||
quality: int) -> Response:
|
||||
data = dict(citizen_id=citizen_id, amount=amount, industry_id=industry, quality=quality, _token=self.token)
|
||||
return self.post(f"{self.url}/economy/donate-items-action", data=data,
|
||||
headers={"Referer": f"{self.url}/economy/donate-items/{citizen_id}"})
|
||||
headers={'Referer': f"{self.url}/economy/donate-items/{citizen_id}"})
|
||||
|
||||
def _post_economy_donate_money_action(self, citizen_id: int, amount: float = 0.0,
|
||||
currency: int = 62) -> Response:
|
||||
data = dict(citizen_id=citizen_id, _token=self.token, currency_id=currency, amount=amount)
|
||||
return self.post(f"{self.url}/economy/donate-money-action", data=data,
|
||||
headers={"Referer": f"{self.url}/economy/donate-money/{citizen_id}"})
|
||||
headers={'Referer': f"{self.url}/economy/donate-money/{citizen_id}"})
|
||||
|
||||
def _post_economy_exchange_purchase(self, amount: float, currency: int, offer: int) -> Response:
|
||||
data = dict(_token=self.token, amount=amount, currencyId=currency, offerId=offer)
|
||||
@ -340,19 +344,19 @@ class ErepublikEconomyAPI(CitizenBaseAPI):
|
||||
def _post_economy_marketplace(self, country: int, industry: int, quality: int,
|
||||
order_asc: bool = True) -> Response:
|
||||
data = dict(countryId=country, industryId=industry, quality=quality, ajaxMarket=1,
|
||||
orderBy="price_asc" if order_asc else "price_desc", _token=self.token)
|
||||
orderBy='price_asc' if order_asc else 'price_desc', _token=self.token)
|
||||
return self.post(f"{self.url}/economy/marketplaceAjax", data=data)
|
||||
|
||||
def _post_economy_marketplace_actions(self, action: str, **kwargs) -> Response:
|
||||
if action == 'buy':
|
||||
data = dict(_token=self.token, offerId=kwargs['offer'], amount=kwargs['amount'],
|
||||
orderBy="price_asc", currentPage=1, buyAction=1)
|
||||
orderBy='price_asc', currentPage=1, buyAction=1)
|
||||
elif action == 'sell':
|
||||
data = dict(_token=self.token, countryId=kwargs["country_id"], price=kwargs["price"],
|
||||
industryId=kwargs["industry"], quality=kwargs["quality"], amount=kwargs['amount'],
|
||||
data = dict(_token=self.token, countryId=kwargs['country_id'], price=kwargs['price'],
|
||||
industryId=kwargs['industry'], quality=kwargs['quality'], amount=kwargs['amount'],
|
||||
sellAction='postOffer')
|
||||
elif action == 'delete':
|
||||
data = dict(_token=self.token, offerId=kwargs["offer_id"], sellAction='deleteOffer')
|
||||
data = dict(_token=self.token, offerId=kwargs['offer_id'], sellAction='deleteOffer')
|
||||
else:
|
||||
raise ValueError(f"Action '{action}' is not supported! Only 'buy/sell/delete' actions are available")
|
||||
return self.post(f"{self.url}/economy/marketplaceActions", data=data)
|
||||
@ -380,7 +384,7 @@ class ErepublikLocationAPI(CitizenBaseAPI):
|
||||
def _get_main_city_data_residents(self, city_id: int, page: int = 1, params: Mapping[str, Any] = None) -> Response:
|
||||
if params is None:
|
||||
params = {}
|
||||
return self.get(f"{self.url}/main/city-data/{city_id}/residents", params={"currentPage": page, **params})
|
||||
return self.get(f"{self.url}/main/city-data/{city_id}/residents", params={'currentPage': page, **params})
|
||||
|
||||
|
||||
class ErepublikMilitaryAPI(CitizenBaseAPI):
|
||||
@ -403,7 +407,7 @@ class ErepublikMilitaryAPI(CitizenBaseAPI):
|
||||
return self.get(f"{self.url}/military/campaignsJson/citizen")
|
||||
|
||||
def _get_military_unit_data(self, unit_id: int, **kwargs) -> Response:
|
||||
params = {"groupId": unit_id, "panel": "members", **kwargs}
|
||||
params = {'groupId': unit_id, 'panel': 'members', **kwargs}
|
||||
return self.get(f"{self.url}/military/military-unit-data/", params=params)
|
||||
|
||||
def _post_main_activate_battle_effect(self, battle_id: int, kind: str, citizen_id: int) -> Response:
|
||||
@ -431,10 +435,10 @@ class ErepublikMilitaryAPI(CitizenBaseAPI):
|
||||
|
||||
def _post_military_battle_console(self, battle_id: int, action: str, page: int = 1, **kwargs) -> Response:
|
||||
data = dict(battleId=battle_id, action=action, _token=self.token)
|
||||
if action == "battleStatistics":
|
||||
data.update(round=kwargs["round_id"], zoneId=kwargs["round_id"], leftPage=page, rightPage=page,
|
||||
division=kwargs["division"], type=kwargs.get("type", 'damage'), )
|
||||
elif action == "warList":
|
||||
if action == 'battleStatistics':
|
||||
data.update(round=kwargs['round_id'], zoneId=kwargs['round_id'], leftPage=page, rightPage=page,
|
||||
division=kwargs['division'], type=kwargs.get('type', 'damage'), )
|
||||
elif action == 'warList':
|
||||
data.update(page=page)
|
||||
return self.post(f"{self.url}/military/battle-console", data=data)
|
||||
|
||||
@ -451,9 +455,39 @@ class ErepublikMilitaryAPI(CitizenBaseAPI):
|
||||
data = dict(sideId=side_id, battleId=battle_id, _token=self.token, battleZoneId=zone_id)
|
||||
return self.post(f"{self.url}/military/fight-shooot/{battle_id}", data=data)
|
||||
|
||||
def _post_fight_deploy_deploy_report_data(self, deployment_id: int):
|
||||
def _post_fight_deploy_deploy_report_data(self, deployment_id: int) -> Response:
|
||||
data = dict(_token=self.token, deploymentId=deployment_id)
|
||||
return self.post(f"{self.url}/military/fightDeploy-deployReportData", json=data)
|
||||
return self.post(f"{self.url}/military/fightDeploy-deployReportData", data=data)
|
||||
|
||||
def _post_fight_deploy_get_inventory(self, battle_id: int, side_id: int, battle_zone_id: int) -> Response:
|
||||
data = dict(_token=self.token, battleId=battle_id, sideCountryId=side_id, battleZoneId=battle_zone_id)
|
||||
return self.post(f"{self.url}/military/fightDeploy-getInventory", data=data)
|
||||
|
||||
def _post_fight_deploy_start_deploy(
|
||||
self, battle_id: int, side_id: int, battle_zone_id: int, energy: int, weapon: int, **kwargs
|
||||
) -> Response:
|
||||
data = dict(_token=self.token, battleId=battle_id, battleZoneId=battle_zone_id, sideCountryId=side_id,
|
||||
weaponQuality=weapon, totalEnergy=energy, **kwargs)
|
||||
return self.post(f"{self.url}/military/fightDeploy-startDeploy", data=data)
|
||||
|
||||
def _get_main_session_captcha(self) -> Response:
|
||||
return self.get(f'{self.url}/main/sessionCaptcha')
|
||||
|
||||
def _get_main_session_unlock_popup(self) -> Response:
|
||||
return self.get(f'{self.url}/main/sessionUnlockPopup')
|
||||
|
||||
def _post_main_session_get_challenge(self, captcha_id: int) -> Response:
|
||||
env = dict(l=['tets', ], s=[], c=["l_chathwe", "l_chatroom"], m=0)
|
||||
data = dict(_token=self.token, captchaId=captcha_id, env=utils.b64json(env))
|
||||
return self.post(f'{self.url}/main/sessionGetChallenge', data=data)
|
||||
|
||||
def _post_main_session_unlock(
|
||||
self, captcha: int, image: str, challenge: str, coords: List[Dict[str, int]], src: str
|
||||
) -> Response:
|
||||
env = dict(l=['tets', ], s=[], c=["l_chathwe", "l_chatroom"], m=0)
|
||||
data = dict(_token=self.token, captchaId=captcha, imageId=image, challengeId=challenge,
|
||||
clickMatrix=utils.json.dumps(coords), isMobile=0, env=utils.b64json(env), src=src)
|
||||
return self.post(f'{self.url}/main/sessionUnlock', data=data)
|
||||
|
||||
|
||||
class ErepublikPoliticsAPI(CitizenBaseAPI):
|
||||
@ -514,10 +548,10 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
return self.get(f"{self.url}/main/messages-paginated/{page}")
|
||||
|
||||
def _get_main_money_donation_accept(self, donation_id: int) -> Response:
|
||||
return self.get(f"{self.url}/main/money-donation/accept/{donation_id}", params={"_token": self.token})
|
||||
return self.get(f"{self.url}/main/money-donation/accept/{donation_id}", params={'_token': self.token})
|
||||
|
||||
def _get_main_money_donation_reject(self, donation_id: int) -> Response:
|
||||
return self.get(f"{self.url}/main/money-donation/reject/{donation_id}", params={"_token": self.token})
|
||||
return self.get(f"{self.url}/main/money-donation/reject/{donation_id}", params={'_token': self.token})
|
||||
|
||||
def _get_main_notifications_ajax_community(self, page: int = 1) -> Response:
|
||||
return self.get(f"{self.url}/main/notificationsAjax/community/{page}")
|
||||
@ -537,16 +571,16 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
def _post_main_citizen_add_remove_friend(self, citizen: int, add: bool) -> Response:
|
||||
data = dict(_token=self.token, citizenId=citizen, url="//www.erepublik.com/en/main/citizen-addRemoveFriend")
|
||||
if add:
|
||||
data.update({"action": "addFriend"})
|
||||
data.update({'action': 'addFriend'})
|
||||
else:
|
||||
data.update({"action": "removeFriend"})
|
||||
data.update({'action': 'removeFriend'})
|
||||
return self.post(f"{self.url}/main/citizen-addRemoveFriend", data=data)
|
||||
|
||||
def _post_main_daily_task_reward(self) -> Response:
|
||||
return self.post(f"{self.url}/main/daily-tasks-reward", data=dict(_token=self.token))
|
||||
|
||||
def _post_delete_message(self, msg_id: list) -> Response:
|
||||
data = {"_token": self.token, "delete_message[]": msg_id}
|
||||
data = {'_token': self.token, "delete_message[]": msg_id}
|
||||
return self.post(f"{self.url}/main/messages-delete", data)
|
||||
|
||||
def _post_eat(self, color: str) -> Response:
|
||||
@ -558,7 +592,7 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/main/global-alerts/close", data=data)
|
||||
|
||||
def _post_forgot_password(self, email: str) -> Response:
|
||||
data = dict(_token=self.token, email=email, commit="Reset password")
|
||||
data = dict(_token=self.token, email=email, commit='Reset password')
|
||||
return self.post(f"{self.url}/forgot-password", data=data)
|
||||
|
||||
def _post_login(self, email: str, password: str) -> Response:
|
||||
@ -566,19 +600,19 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/login", data=data)
|
||||
|
||||
def _post_main_messages_alert(self, notification_ids: List[int]) -> Response:
|
||||
data = {"_token": self.token, "delete_alerts[]": notification_ids, "deleteAllAlerts": "1", "delete": "Delete"}
|
||||
data = {'_token': self.token, "delete_alerts[]": notification_ids, 'deleteAllAlerts': '1', 'delete': 'Delete'}
|
||||
return self.post(f"{self.url}/main/messages-alerts/1", data=data)
|
||||
|
||||
def _post_main_notifications_ajax_community(self, notification_ids: List[int], page: int = 1) -> Response:
|
||||
data = {"_token": self.token, "delete_alerts[]": notification_ids}
|
||||
data = {'_token': self.token, "delete_alerts[]": notification_ids}
|
||||
return self.post(f"{self.url}/main/notificationsAjax/community/{page}", data=data)
|
||||
|
||||
def _post_main_notifications_ajax_system(self, notification_ids: List[int], page: int = 1) -> Response:
|
||||
data = {"_token": self.token, "delete_alerts[]": notification_ids}
|
||||
data = {'_token': self.token, "delete_alerts[]": notification_ids}
|
||||
return self.post(f"{self.url}/main/notificationsAjax/system/{page}", data=data)
|
||||
|
||||
def _post_main_notifications_ajax_report(self, notification_ids: List[int], page: int = 1) -> Response:
|
||||
data = {"_token": self.token, "delete_alerts[]": notification_ids}
|
||||
data = {'_token': self.token, "delete_alerts[]": notification_ids}
|
||||
return self.post(f"{self.url}/main/notificationsAjax/report/{page}", data=data)
|
||||
|
||||
def _post_main_messages_compose(self, subject: str, body: str, citizens: List[int]) -> Response:
|
||||
@ -588,7 +622,7 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/main/messages-compose/{url_pk}", data=data)
|
||||
|
||||
def _post_military_group_missions(self) -> Response:
|
||||
data = dict(action="check", _token=self.token)
|
||||
data = dict(action='check', _token=self.token)
|
||||
return self.post(f"{self.url}/military/group-missions", data=data)
|
||||
|
||||
def _post_main_weekly_challenge_reward(self, reward_id: int) -> Response:
|
||||
@ -600,7 +634,7 @@ class ErepublikProfileAPI(CitizenBaseAPI):
|
||||
return self.post(f"{self.url}/main/weekly-challenge-collect-all", data=data)
|
||||
|
||||
def _post_main_profile_update(self, action: str, params: str):
|
||||
data = {"action": action, "params": params, "_token": self.token}
|
||||
data = {'action': action, 'params': params, '_token': self.token}
|
||||
return self.post(f"{self.url}/main/profile-update", data=data)
|
||||
|
||||
|
||||
@ -617,73 +651,73 @@ class ErepublikWallPostAPI(CitizenBaseAPI):
|
||||
# ## Country
|
||||
|
||||
def _post_main_country_comment_retrieve(self, post_id: int) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id}
|
||||
data = {'_token': self.token, 'postId': post_id}
|
||||
return self.post(f"{self.url}/main/country-comment/retrieve/json", data=data)
|
||||
|
||||
def _post_main_country_comment_create(self, post_id: int, comment_message: str) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id, 'comment_message': comment_message}
|
||||
data = {'_token': self.token, 'postId': post_id, 'comment_message': comment_message}
|
||||
return self.post(f"{self.url}/main/country-comment/create/json", data=data)
|
||||
|
||||
def _post_main_country_post_create(self, body: str, post_as: int) -> Response:
|
||||
data = {"_token": self.token, "post_message": body, "post_as": post_as}
|
||||
data = {'_token': self.token, 'post_message': body, 'post_as': post_as}
|
||||
return self.post(f"{self.url}/main/country-post/create/json", data=data)
|
||||
|
||||
def _post_main_country_post_retrieve(self) -> Response:
|
||||
data = {"_token": self.token, "page": 1, "switchedFrom": False}
|
||||
data = {'_token': self.token, 'page': 1, 'switchedFrom': False}
|
||||
return self.post(f"{self.url}/main/country-post/retrieve/json", data=data)
|
||||
|
||||
# ## Military Unit
|
||||
|
||||
def _post_main_military_unit_comment_retrieve(self, post_id: int) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id}
|
||||
data = {'_token': self.token, 'postId': post_id}
|
||||
return self.post(f"{self.url}/main/military-unit-comment/retrieve/json", data=data)
|
||||
|
||||
def _post_main_military_unit_comment_create(self, post_id: int, comment_message: str) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id, 'comment_message': comment_message}
|
||||
data = {'_token': self.token, 'postId': post_id, 'comment_message': comment_message}
|
||||
return self.post(f"{self.url}/main/military-unit-comment/create/json", data=data)
|
||||
|
||||
def _post_main_military_unit_post_create(self, body: str, post_as: int) -> Response:
|
||||
data = {"_token": self.token, "post_message": body, "post_as": post_as}
|
||||
data = {'_token': self.token, 'post_message': body, 'post_as': post_as}
|
||||
return self.post(f"{self.url}/main/military-unit-post/create/json", data=data)
|
||||
|
||||
def _post_main_military_unit_post_retrieve(self) -> Response:
|
||||
data = {"_token": self.token, "page": 1, "switchedFrom": False}
|
||||
data = {'_token': self.token, 'page': 1, 'switchedFrom': False}
|
||||
return self.post(f"{self.url}/main/military-unit-post/retrieve/json", data=data)
|
||||
|
||||
# ## Party
|
||||
|
||||
def _post_main_party_comment_retrieve(self, post_id: int) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id}
|
||||
data = {'_token': self.token, 'postId': post_id}
|
||||
return self.post(f"{self.url}/main/party-comment/retrieve/json", data=data)
|
||||
|
||||
def _post_main_party_comment_create(self, post_id: int, comment_message: str) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id, 'comment_message': comment_message}
|
||||
data = {'_token': self.token, 'postId': post_id, 'comment_message': comment_message}
|
||||
return self.post(f"{self.url}/main/party-comment/create/json", data=data)
|
||||
|
||||
def _post_main_party_post_create(self, body: str) -> Response:
|
||||
data = {"_token": self.token, "post_message": body}
|
||||
data = {'_token': self.token, 'post_message': body}
|
||||
return self.post(f"{self.url}/main/party-post/create/json", data=data)
|
||||
|
||||
def _post_main_party_post_retrieve(self) -> Response:
|
||||
data = {"_token": self.token, "page": 1, "switchedFrom": False}
|
||||
data = {'_token': self.token, 'page': 1, 'switchedFrom': False}
|
||||
return self.post(f"{self.url}/main/party-post/retrieve/json", data=data)
|
||||
|
||||
# ## Friend's Wall
|
||||
|
||||
def _post_main_wall_comment_retrieve(self, post_id: int) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id}
|
||||
data = {'_token': self.token, 'postId': post_id}
|
||||
return self.post(f"{self.url}/main/wall-comment/retrieve/json", data=data)
|
||||
|
||||
def _post_main_wall_comment_create(self, post_id: int, comment_message: str) -> Response:
|
||||
data = {"_token": self.token, "postId": post_id, 'comment_message': comment_message}
|
||||
data = {'_token': self.token, 'postId': post_id, 'comment_message': comment_message}
|
||||
return self.post(f"{self.url}/main/wall-comment/create/json", data=data)
|
||||
|
||||
def _post_main_wall_post_create(self, body: str) -> Response:
|
||||
data = {"_token": self.token, "post_message": body}
|
||||
data = {'_token': self.token, 'post_message': body}
|
||||
return self.post(f"{self.url}/main/wall-post/create/json", data=data)
|
||||
|
||||
def _post_main_wall_post_retrieve(self) -> Response:
|
||||
data = {"_token": self.token, "page": 1, "switchedFrom": False}
|
||||
data = {'_token': self.token, 'page': 1, 'switchedFrom': False}
|
||||
return self.post(f"{self.url}/main/wall-post/retrieve/json", data=data)
|
||||
|
||||
# ## Medal posting
|
||||
|
File diff suppressed because it is too large
Load Diff
@ -3,16 +3,15 @@ import hashlib
|
||||
import threading
|
||||
import weakref
|
||||
from decimal import Decimal
|
||||
from typing import Any, Dict, Generator, Iterable, List, NamedTuple, NoReturn, Tuple, Union
|
||||
from typing import Any, Dict, Generator, Iterable, List, NamedTuple, NoReturn, Union
|
||||
|
||||
from requests import Response, Session, post
|
||||
|
||||
from . import constants, types, utils
|
||||
|
||||
__all__ = ['Battle', 'BattleDivision', 'BattleSide', 'Company', 'Config', 'Details', 'Energy', 'ErepublikException',
|
||||
'ErepublikNetworkException', 'EnergyToFight',
|
||||
'Holding', 'MyCompanies', 'MyJSONEncoder', 'OfferItem', 'Politics', 'Reporter', 'TelegramReporter',
|
||||
'Inventory']
|
||||
'ErepublikJSONEncoder', 'ErepublikNetworkException', 'EnergyToFight', 'Holding', 'Inventory', 'MyCompanies',
|
||||
'OfferItem', 'Politics', 'Reporter', 'TelegramReporter', ]
|
||||
|
||||
|
||||
class ErepublikException(Exception):
|
||||
@ -29,7 +28,7 @@ class ErepublikNetworkException(ErepublikException):
|
||||
class Holding:
|
||||
id: int
|
||||
region: int
|
||||
companies: List["Company"]
|
||||
companies: List['Company']
|
||||
name: str
|
||||
_citizen = weakref.ReferenceType
|
||||
|
||||
@ -37,16 +36,16 @@ class Holding:
|
||||
self._citizen = weakref.ref(citizen)
|
||||
self.id: int = _id
|
||||
self.region: int = region
|
||||
self.companies: List["Company"] = list()
|
||||
self.companies: List['Company'] = list()
|
||||
if name:
|
||||
self.name = name
|
||||
else:
|
||||
comp_sum = len(self.companies)
|
||||
name = f"Holding (#{self.id}) with {comp_sum} "
|
||||
if comp_sum == 1:
|
||||
name += "company"
|
||||
name += 'company'
|
||||
else:
|
||||
name += "companies"
|
||||
name += 'companies'
|
||||
self.name = name
|
||||
|
||||
@property
|
||||
@ -54,20 +53,20 @@ class Holding:
|
||||
return len([1 for company in self.companies if company.wam_enabled and not company.already_worked])
|
||||
|
||||
@property
|
||||
def wam_companies(self) -> Iterable["Company"]:
|
||||
def wam_companies(self) -> Iterable['Company']:
|
||||
return [company for company in self.companies if company.wam_enabled]
|
||||
|
||||
@property
|
||||
def employable_companies(self) -> Iterable["Company"]:
|
||||
def employable_companies(self) -> Iterable['Company']:
|
||||
return [company for company in self.companies if company.preset_works]
|
||||
|
||||
def add_company(self, company: "Company") -> NoReturn:
|
||||
def add_company(self, company: 'Company') -> NoReturn:
|
||||
self.companies.append(company)
|
||||
self.companies.sort()
|
||||
|
||||
def get_wam_raw_usage(self) -> Dict[str, Decimal]:
|
||||
frm = Decimal("0.00")
|
||||
wrm = Decimal("0.00")
|
||||
frm = Decimal('0.00')
|
||||
wrm = Decimal('0.00')
|
||||
for company in self.wam_companies:
|
||||
if company.industry in [1, 7, 8, 9, 10, 11]:
|
||||
frm += company.raw_usage
|
||||
@ -75,11 +74,11 @@ class Holding:
|
||||
wrm += company.raw_usage
|
||||
return dict(frm=frm, wrm=wrm)
|
||||
|
||||
def get_wam_companies(self, raw_factory: bool = None) -> List["Company"]:
|
||||
def get_wam_companies(self, raw_factory: bool = None) -> List['Company']:
|
||||
raw = []
|
||||
factory = []
|
||||
for company in self.wam_companies:
|
||||
if not company.already_worked and not company.cannot_wam_reason == "war":
|
||||
if not company.already_worked and not company.cannot_wam_reason == 'war':
|
||||
if company.is_raw:
|
||||
raw.append(company)
|
||||
else:
|
||||
@ -93,9 +92,9 @@ class Holding:
|
||||
comp = len(self.companies)
|
||||
name = f"Holding (#{self.id}) with {comp} "
|
||||
if comp == 1:
|
||||
name += "company"
|
||||
name += 'company'
|
||||
else:
|
||||
name += "companies"
|
||||
name += 'companies'
|
||||
return name
|
||||
|
||||
def __repr__(self):
|
||||
@ -147,10 +146,10 @@ class Company:
|
||||
self.raw_usage = - self.products_made * raw_usage
|
||||
|
||||
def _get_real_quality(self, quality) -> int:
|
||||
# 7: "FRM q1", 8: "FRM q2", 9: "FRM q3", 10: "FRM q4", 11: "FRM q5",
|
||||
# 12: "WRM q1", 13: "WRM q2", 14: "WRM q3", 15: "WRM q4", 16: "WRM q5",
|
||||
# 18: "HRM q1", 19: "HRM q2", 20: "HRM q3", 21: "HRM q4", 22: "HRM q5",
|
||||
# 24: "ARM q1", 25: "ARM q2", 26: "ARM q3", 27: "ARM q4", 28: "ARM q5",
|
||||
# 7: 'FRM q1', 8: 'FRM q2', 9: 'FRM q3', 10: 'FRM q4', 11: 'FRM q5',
|
||||
# 12: 'WRM q1', 13: 'WRM q2', 14: 'WRM q3', 15: 'WRM q4', 16: 'WRM q5',
|
||||
# 18: 'HRM q1', 19: 'HRM q2', 20: 'HRM q3', 21: 'HRM q4', 22: 'HRM q5',
|
||||
# 24: 'ARM q1', 25: 'ARM q2', 26: 'ARM q3', 27: 'ARM q4', 28: 'ARM q5',
|
||||
if 7 <= self.industry <= 11:
|
||||
return self.industry % 6
|
||||
elif 12 <= self.industry <= 16:
|
||||
@ -164,10 +163,10 @@ class Company:
|
||||
|
||||
@property
|
||||
def _internal_industry(self) -> int:
|
||||
# 7: "FRM q1", 8: "FRM q2", 9: "FRM q3", 10: "FRM q4", 11: "FRM q5",
|
||||
# 12: "WRM q1", 13: "WRM q2", 14: "WRM q3", 15: "WRM q4", 16: "WRM q5",
|
||||
# 18: "HRM q1", 19: "HRM q2", 20: "HRM q3", 21: "HRM q4", 22: "HRM q5",
|
||||
# 24: "ARM q1", 25: "ARM q2", 26: "ARM q3", 27: "ARM q4", 28: "ARM q5",
|
||||
# 7: 'FRM q1', 8: 'FRM q2', 9: 'FRM q3', 10: 'FRM q4', 11: 'FRM q5',
|
||||
# 12: 'WRM q1', 13: 'WRM q2', 14: 'WRM q3', 15: 'WRM q4', 16: 'WRM q5',
|
||||
# 18: 'HRM q1', 19: 'HRM q2', 20: 'HRM q3', 21: 'HRM q4', 22: 'HRM q5',
|
||||
# 24: 'ARM q1', 25: 'ARM q2', 26: 'ARM q3', 27: 'ARM q4', 28: 'ARM q5',
|
||||
if 7 <= self.industry <= 11:
|
||||
return 7
|
||||
elif 12 <= self.industry <= 16:
|
||||
@ -186,22 +185,22 @@ class Company:
|
||||
def __hash__(self):
|
||||
return hash(self._sort_keys)
|
||||
|
||||
def __lt__(self, other: "Company"):
|
||||
def __lt__(self, other: 'Company'):
|
||||
return self._sort_keys < other._sort_keys
|
||||
|
||||
def __le__(self, other: "Company"):
|
||||
def __le__(self, other: 'Company'):
|
||||
return self._sort_keys <= other._sort_keys
|
||||
|
||||
def __gt__(self, other: "Company"):
|
||||
def __gt__(self, other: 'Company'):
|
||||
return self._sort_keys > other._sort_keys
|
||||
|
||||
def __ge__(self, other: "Company"):
|
||||
def __ge__(self, other: 'Company'):
|
||||
return self._sort_keys >= other._sort_keys
|
||||
|
||||
def __eq__(self, other: "Company"):
|
||||
def __eq__(self, other: 'Company'):
|
||||
return self._sort_keys == other._sort_keys
|
||||
|
||||
def __ne__(self, other: "Company"):
|
||||
def __ne__(self, other: 'Company'):
|
||||
return self._sort_keys != other._sort_keys
|
||||
|
||||
def __str__(self):
|
||||
@ -468,6 +467,7 @@ class Details:
|
||||
pp: int = 0
|
||||
pin: str = None
|
||||
gold: float = 0
|
||||
level: int = 0
|
||||
next_pp: List[int] = None
|
||||
citizen_id: int = 0
|
||||
citizenship: constants.Country
|
||||
@ -512,11 +512,15 @@ class Details:
|
||||
@property
|
||||
def as_dict(self) -> Dict[str, Union[int, float, str, constants.Country, bool]]:
|
||||
return dict(xp=self.xp, cc=self.cc, pp=self.pp, pin=self.pin, gold=self.gold, next_pp=self.next_pp,
|
||||
citizen_id=self.citizen_id, citizenship=self.citizenship, current_region=self.current_region,
|
||||
current_country=self.current_country, residence_region=self.residence_region,
|
||||
residence_country=self.residence_country, daily_task_done=self.daily_task_done,
|
||||
daily_task_reward=self.daily_task_reward, mayhem_skills=self.mayhem_skills,
|
||||
xp_till_level_up=self.xp_till_level_up)
|
||||
level=self.level, citizen_id=self.citizen_id, citizenship=self.citizenship,
|
||||
current_region=self.current_region, current_country=self.current_country,
|
||||
residence_region=self.residence_region, residence_country=self.residence_country,
|
||||
daily_task_done=self.daily_task_done, daily_task_reward=self.daily_task_reward,
|
||||
mayhem_skills=self.mayhem_skills, xp_till_level_up=self.xp_till_level_up)
|
||||
|
||||
@property
|
||||
def is_elite(self):
|
||||
return self.level > 100
|
||||
|
||||
|
||||
class Politics:
|
||||
@ -575,7 +579,7 @@ class Reporter:
|
||||
self._citizen = weakref.ref(citizen)
|
||||
self._req = Session()
|
||||
self.url = "https://api.erep.lv"
|
||||
self._req.headers.update({"user-agent": "eRepublik Script Reporter v3",
|
||||
self._req.headers.update({"user-agent": 'eRepublik Script Reporter v3',
|
||||
'erep-version': utils.__version__,
|
||||
'erep-user-id': str(self.citizen_id),
|
||||
'erep-user-name': self.citizen.name})
|
||||
@ -599,10 +603,10 @@ class Reporter:
|
||||
if self.__to_update:
|
||||
for unreported_data in self.__to_update:
|
||||
unreported_data.update(player_id=self.citizen_id, key=self.key)
|
||||
unreported_data = utils.json.loads(utils.json.dumps(unreported_data, cls=MyJSONEncoder))
|
||||
unreported_data = utils.json.loads(utils.json.dumps(unreported_data, cls=ErepublikJSONEncoder))
|
||||
self._req.post(f"{self.url}/bot/update", json=unreported_data)
|
||||
self.__to_update.clear()
|
||||
data = utils.json.loads(utils.json.dumps(data, cls=MyJSONEncoder))
|
||||
data = utils.json.loads(utils.json.dumps(data, cls=ErepublikJSONEncoder))
|
||||
r = self._req.post(f"{self.url}/bot/update", json=data)
|
||||
return r
|
||||
|
||||
@ -610,13 +614,13 @@ class Reporter:
|
||||
if not self.__registered:
|
||||
try:
|
||||
r = self.__bot_update(dict(key=self.key, check=True, player_id=self.citizen_id))
|
||||
if not r.json().get("status"):
|
||||
if not r.json().get('status'):
|
||||
self._req.post(f"{self.url}/bot/register", json=dict(name=self.name, email=self.email,
|
||||
player_id=self.citizen_id))
|
||||
finally:
|
||||
self.__registered = True
|
||||
self.allowed = True
|
||||
self.report_action("STARTED", value=utils.now().strftime("%F %T"))
|
||||
self.report_action('STARTED', value=utils.now().strftime("%F %T"))
|
||||
|
||||
def send_state_update(self, xp: int, cc: float, gold: float, inv_total: int, inv: int,
|
||||
hp_limit: int, hp_interval: int, hp_available: int, food: int, pp: int):
|
||||
@ -644,7 +648,7 @@ class Reporter:
|
||||
else:
|
||||
self.__to_update.append(json_data)
|
||||
|
||||
def report_fighting(self, battle: "Battle", invader: bool, division: "BattleDivision", damage: float, hits: int):
|
||||
def report_fighting(self, battle: 'Battle', invader: bool, division: 'BattleDivision', damage: float, hits: int):
|
||||
side = battle.invader if invader else battle.defender
|
||||
self.report_action('FIGHT', dict(battle_id=battle.id, side=side, dmg=damage,
|
||||
air=battle.has_air, hits=hits,
|
||||
@ -663,7 +667,7 @@ class Reporter:
|
||||
def report_promo(self, kind: str, time_until: datetime.datetime):
|
||||
self._req.post(f"{self.url}/promos/add/", data=dict(kind=kind, time_untill=time_until))
|
||||
|
||||
def fetch_battle_priorities(self, country: constants.Country) -> List["Battle"]:
|
||||
def fetch_battle_priorities(self, country: constants.Country) -> List['Battle']:
|
||||
try:
|
||||
battle_response = self._req.get(f'{self.url}/api/v1/battles/{country.id}')
|
||||
return [self.citizen.all_battles[bid] for bid in battle_response.json().get('battle_ids', []) if
|
||||
@ -671,16 +675,19 @@ class Reporter:
|
||||
except: # noqa
|
||||
return []
|
||||
|
||||
def fetch_tasks(self) -> List[Union[str, Tuple[Any]]]:
|
||||
def fetch_tasks(self) -> List[Dict[str, Any]]:
|
||||
try:
|
||||
task_response = self._req.get(f'{self.url}/api/v1/command',
|
||||
params=dict(citizen=self.citizen_id, key=self.key))
|
||||
return task_response.json().get('task_collection')
|
||||
task_response = self._req.post(
|
||||
f'{self.url}/api/v1/command', data=dict(citizen=self.citizen_id, key=self.key)).json()
|
||||
if task_response.get('status'):
|
||||
return task_response.get('data')
|
||||
else:
|
||||
return []
|
||||
except: # noqa
|
||||
return []
|
||||
|
||||
|
||||
class MyJSONEncoder(utils.json.JSONEncoder):
|
||||
class ErepublikJSONEncoder(utils.json.JSONEncoder):
|
||||
def default(self, o):
|
||||
from erepublik.citizen import Citizen
|
||||
if isinstance(o, Decimal):
|
||||
@ -694,7 +701,7 @@ class MyJSONEncoder(utils.json.JSONEncoder):
|
||||
return dict(__type__='timedelta', days=o.days, seconds=o.seconds,
|
||||
microseconds=o.microseconds, total_seconds=o.total_seconds())
|
||||
elif isinstance(o, Response):
|
||||
return dict(headers=o.headers.__dict__, url=o.url, text=o.text)
|
||||
return dict(headers=dict(o.__dict__['headers']), url=o.url, text=o.text, status_code=o.status_code)
|
||||
elif hasattr(o, 'as_dict'):
|
||||
return o.as_dict
|
||||
elif isinstance(o, set):
|
||||
@ -711,12 +718,12 @@ class BattleSide:
|
||||
points: int
|
||||
deployed: List[constants.Country]
|
||||
allies: List[constants.Country]
|
||||
battle: "Battle"
|
||||
battle: 'Battle'
|
||||
_battle: weakref.ReferenceType
|
||||
country: constants.Country
|
||||
is_defender: bool
|
||||
|
||||
def __init__(self, battle: "Battle", country: constants.Country, points: int, allies: List[constants.Country],
|
||||
def __init__(self, battle: 'Battle', country: constants.Country, points: int, allies: List[constants.Country],
|
||||
deployed: List[constants.Country], defender: bool):
|
||||
self._battle = weakref.ref(battle)
|
||||
self.country = country
|
||||
@ -730,11 +737,11 @@ class BattleSide:
|
||||
return self.country.id
|
||||
|
||||
def __repr__(self):
|
||||
side_text = "Defender" if self.is_defender else "Invader "
|
||||
side_text = 'Defender' if self.is_defender else 'Invader '
|
||||
return f"<BattleSide: {side_text} {self.country.name}|{self.points:>2d}p>"
|
||||
|
||||
def __str__(self):
|
||||
side_text = "Defender" if self.is_defender else "Invader "
|
||||
side_text = 'Defender' if self.is_defender else 'Invader '
|
||||
return f"{side_text} {self.country.name} - {self.points:>2d} points"
|
||||
|
||||
def __format__(self, format_spec):
|
||||
@ -760,7 +767,7 @@ class BattleDivision:
|
||||
inv_medal: Dict[str, int]
|
||||
terrain: int
|
||||
div: int
|
||||
battle: "Battle"
|
||||
battle: 'Battle'
|
||||
_battle: weakref.ReferenceType
|
||||
|
||||
@property
|
||||
@ -776,7 +783,7 @@ class BattleDivision:
|
||||
def div_end(self) -> bool:
|
||||
return utils.now() >= self.end
|
||||
|
||||
def __init__(self, battle: "Battle", div_id: int, end: datetime.datetime, epic: bool, div: int, wall_for: int,
|
||||
def __init__(self, battle: 'Battle', div_id: int, end: datetime.datetime, epic: bool, div: int, wall_for: int,
|
||||
wall_dom: float, terrain_id: int = 0):
|
||||
"""Battle division helper class
|
||||
|
||||
@ -792,7 +799,7 @@ class BattleDivision:
|
||||
self.id = div_id
|
||||
self.end = end
|
||||
self.epic = epic
|
||||
self.wall = dict({"for": wall_for, "dom": wall_dom})
|
||||
self.wall = {'for': wall_for, 'dom': wall_dom}
|
||||
self.terrain = terrain_id
|
||||
self.div = div
|
||||
|
||||
@ -805,7 +812,7 @@ class BattleDivision:
|
||||
if self.terrain:
|
||||
base_name += f" ({self.terrain_display})"
|
||||
if self.div_end:
|
||||
base_name += " Ended"
|
||||
base_name += ' Ended'
|
||||
return base_name
|
||||
|
||||
def __repr__(self):
|
||||
@ -897,8 +904,8 @@ class Battle:
|
||||
|
||||
battle_div = BattleDivision(self, div_id=data.get('id'), div=data.get('div'), end=end,
|
||||
epic=data.get('epic_type') in [1, 5],
|
||||
wall_for=data.get('wall').get("for"),
|
||||
wall_dom=data.get('wall').get("dom"),
|
||||
wall_for=data.get('wall').get('for'),
|
||||
wall_dom=data.get('wall').get('dom'),
|
||||
terrain_id=data.get('terrain', 0))
|
||||
|
||||
self.div.update({div: battle_div})
|
||||
@ -984,7 +991,7 @@ class TelegramReporter:
|
||||
self._last_time = utils.good_timedelta(utils.now(), datetime.timedelta(minutes=-5))
|
||||
self._last_full_energy_report = utils.good_timedelta(utils.now(), datetime.timedelta(minutes=-30))
|
||||
if self.__queue:
|
||||
self.send_message("Telegram initialized")
|
||||
self.send_message('Telegram initialized')
|
||||
|
||||
def send_message(self, message: str) -> bool:
|
||||
self.__queue.append(message)
|
||||
@ -1012,7 +1019,7 @@ class TelegramReporter:
|
||||
new_line = '\n' if multiple else ''
|
||||
self.send_message(f"New award: {new_line}*{msg}*")
|
||||
|
||||
def report_fight(self, battle: "Battle", invader: bool, division: "BattleDivision", damage: float, hits: int):
|
||||
def report_fight(self, battle: 'Battle', invader: bool, division: 'BattleDivision', damage: float, hits: int):
|
||||
side_txt = (battle.invader if invader else battle.defender).country.iso
|
||||
self.send_message(f"*Fight report*:\n{int(damage):,d} dmg ({hits} hits) in"
|
||||
f" [battle {battle.id} for {battle.region_name[:16]}]({battle.link}) in d{division.div} on "
|
||||
@ -1035,7 +1042,7 @@ class TelegramReporter:
|
||||
message = "\n\n".join(self.__queue)
|
||||
if self.player_name:
|
||||
message = f"Player *{self.player_name}*\n\n" + message
|
||||
response = post(self.api_url, json=dict(chat_id=self.chat_id, text=message, parse_mode="Markdown"))
|
||||
response = post(self.api_url, json=dict(chat_id=self.chat_id, text=message, parse_mode='Markdown'))
|
||||
self._last_time = utils.now()
|
||||
if response.json().get('ok'):
|
||||
self.__queue.clear()
|
||||
|
@ -3,8 +3,8 @@ from typing import Dict, Optional, Union
|
||||
|
||||
import pytz
|
||||
|
||||
__all__ = ["erep_tz", 'min_datetime', "max_datetime", "Country", "AIR_RANKS", "COUNTRIES", "FOOD_ENERGY",
|
||||
"GROUND_RANKS", "GROUND_RANK_POINTS", "INDUSTRIES", "TERRAINS"]
|
||||
__all__ = ['erep_tz', 'min_datetime', 'max_datetime', 'Country', 'AIR_RANKS', 'COUNTRIES', 'FOOD_ENERGY',
|
||||
'GROUND_RANKS', 'GROUND_RANK_POINTS', 'INDUSTRIES', 'TERRAINS']
|
||||
|
||||
erep_tz = pytz.timezone('US/Pacific')
|
||||
min_datetime = erep_tz.localize(datetime.datetime(2007, 11, 20))
|
||||
@ -60,11 +60,11 @@ class Industries:
|
||||
'wrm q1': 12, 'wrm q2': 13, 'wrm q3': 14, 'wrm q4': 15, 'wrm q5': 16,
|
||||
'hrm q1': 18, 'hrm q2': 19, 'hrm q3': 20, 'hrm q4': 21, 'hrm q5': 22,
|
||||
'arm q1': 24, 'arm q2': 25, 'arm q3': 26, 'arm q4': 27, 'arm q5': 28}
|
||||
__by_id = {1: "Food", 2: "Weapon", 3: "Ticket", 4: "House", 23: "Aircraft",
|
||||
7: "foodRaw", 8: "FRM q2", 9: "FRM q3", 10: "FRM q4", 11: "FRM q5",
|
||||
12: "weaponRaw", 13: "WRM q2", 14: "WRM q3", 15: "WRM q4", 16: "WRM q5",
|
||||
17: "houseRaw", 18: "houseRaw", 19: "HRM q2", 20: "HRM q3", 21: "HRM q4", 22: "HRM q5",
|
||||
24: "aircraftRaw", 25: "ARM q2", 26: "ARM q3", 27: "ARM q4", 28: "ARM q5"}
|
||||
__by_id = {1: 'Food', 2: 'Weapon', 3: 'Ticket', 4: 'House', 23: 'Aircraft',
|
||||
7: 'foodRaw', 8: 'FRM q2', 9: 'FRM q3', 10: 'FRM q4', 11: 'FRM q5',
|
||||
12: 'weaponRaw', 13: 'WRM q2', 14: 'WRM q3', 15: 'WRM q4', 16: 'WRM q5',
|
||||
17: 'houseRaw', 18: 'houseRaw', 19: 'HRM q2', 20: 'HRM q3', 21: 'HRM q4', 22: 'HRM q5',
|
||||
24: 'aircraftRaw', 25: 'ARM q2', 26: 'ARM q3', 27: 'ARM q4', 28: 'ARM q5'}
|
||||
|
||||
def __getitem__(self, item) -> Optional[Union[int, str]]:
|
||||
if isinstance(item, int):
|
||||
@ -82,21 +82,21 @@ class Industries:
|
||||
|
||||
|
||||
AIR_RANKS: Dict[int, str] = {
|
||||
1: "Airman", 2: "Airman 1st Class", 3: "Airman 1st Class*", 4: "Airman 1st Class**", 5: "Airman 1st Class***",
|
||||
6: "Airman 1st Class****", 7: "Airman 1st Class*****", 8: "Senior Airman", 9: "Senior Airman*",
|
||||
10: "Senior Airman**", 11: "Senior Airman***", 12: "Senior Airman****", 13: "Senior Airman*****",
|
||||
14: "Staff Sergeant", 15: "Staff Sergeant*", 16: "Staff Sergeant**", 17: "Staff Sergeant***",
|
||||
18: "Staff Sergeant****", 19: "Staff Sergeant*****", 20: "Aviator", 21: "Aviator*", 22: "Aviator**",
|
||||
23: "Aviator***", 24: "Aviator****", 25: "Aviator*****", 26: "Flight Lieutenant", 27: "Flight Lieutenant*",
|
||||
28: "Flight Lieutenant**", 29: "Flight Lieutenant***", 30: "Flight Lieutenant****", 31: "Flight Lieutenant*****",
|
||||
32: "Squadron Leader", 33: "Squadron Leader*", 34: "Squadron Leader**", 35: "Squadron Leader***",
|
||||
36: "Squadron Leader****", 37: "Squadron Leader*****", 38: "Chief Master Sergeant", 39: "Chief Master Sergeant*",
|
||||
40: "Chief Master Sergeant**", 41: "Chief Master Sergeant***", 42: "Chief Master Sergeant****",
|
||||
43: "Chief Master Sergeant*****", 44: "Wing Commander", 45: "Wing Commander*", 46: "Wing Commander**",
|
||||
47: "Wing Commander***", 48: "Wing Commander****", 49: "Wing Commander*****", 50: "Group Captain",
|
||||
51: "Group Captain*", 52: "Group Captain**", 53: "Group Captain***", 54: "Group Captain****",
|
||||
55: "Group Captain*****", 56: "Air Commodore", 57: "Air Commodore*", 58: "Air Commodore**", 59: "Air Commodore***",
|
||||
60: "Air Commodore****", 61: "Air Commodore*****",
|
||||
1: 'Airman', 2: 'Airman 1st Class', 3: 'Airman 1st Class*', 4: 'Airman 1st Class**', 5: 'Airman 1st Class***',
|
||||
6: 'Airman 1st Class****', 7: 'Airman 1st Class*****', 8: 'Senior Airman', 9: 'Senior Airman*',
|
||||
10: 'Senior Airman**', 11: 'Senior Airman***', 12: 'Senior Airman****', 13: 'Senior Airman*****',
|
||||
14: 'Staff Sergeant', 15: 'Staff Sergeant*', 16: 'Staff Sergeant**', 17: 'Staff Sergeant***',
|
||||
18: 'Staff Sergeant****', 19: 'Staff Sergeant*****', 20: 'Aviator', 21: 'Aviator*', 22: 'Aviator**',
|
||||
23: 'Aviator***', 24: 'Aviator****', 25: 'Aviator*****', 26: 'Flight Lieutenant', 27: 'Flight Lieutenant*',
|
||||
28: 'Flight Lieutenant**', 29: 'Flight Lieutenant***', 30: 'Flight Lieutenant****', 31: 'Flight Lieutenant*****',
|
||||
32: 'Squadron Leader', 33: 'Squadron Leader*', 34: 'Squadron Leader**', 35: 'Squadron Leader***',
|
||||
36: 'Squadron Leader****', 37: 'Squadron Leader*****', 38: 'Chief Master Sergeant', 39: 'Chief Master Sergeant*',
|
||||
40: 'Chief Master Sergeant**', 41: 'Chief Master Sergeant***', 42: 'Chief Master Sergeant****',
|
||||
43: 'Chief Master Sergeant*****', 44: 'Wing Commander', 45: 'Wing Commander*', 46: 'Wing Commander**',
|
||||
47: 'Wing Commander***', 48: 'Wing Commander****', 49: 'Wing Commander*****', 50: 'Group Captain',
|
||||
51: 'Group Captain*', 52: 'Group Captain**', 53: 'Group Captain***', 54: 'Group Captain****',
|
||||
55: 'Group Captain*****', 56: 'Air Commodore', 57: 'Air Commodore*', 58: 'Air Commodore**', 59: 'Air Commodore***',
|
||||
60: 'Air Commodore****', 61: 'Air Commodore*****',
|
||||
}
|
||||
|
||||
COUNTRIES: Dict[int, Country] = {
|
||||
@ -145,27 +145,27 @@ COUNTRIES: Dict[int, Country] = {
|
||||
FOOD_ENERGY: Dict[str, int] = dict(q1=2, q2=4, q3=6, q4=8, q5=10, q6=12, q7=20)
|
||||
|
||||
GROUND_RANKS: Dict[int, str] = {
|
||||
1: "Recruit", 2: "Private", 3: "Private*", 4: "Private**", 5: "Private***",
|
||||
6: "Corporal", 7: "Corporal*", 8: "Corporal**", 9: "Corporal***",
|
||||
10: "Sergeant", 11: "Sergeant*", 12: "Sergeant**", 13: "Sergeant***",
|
||||
14: "Lieutenant", 15: "Lieutenant*", 16: "Lieutenant**", 17: "Lieutenant***",
|
||||
18: "Captain", 19: "Captain*", 20: "Captain**", 21: "Captain***",
|
||||
22: "Major", 23: "Major*", 24: "Major**", 25: "Major***",
|
||||
26: "Commander", 27: "Commander*", 28: "Commander**", 29: "Commander***",
|
||||
30: "Lt Colonel", 31: "Lt Colonel*", 32: "Lt Colonel**", 33: "Lt Colonel***",
|
||||
34: "Colonel", 35: "Colonel*", 36: "Colonel**", 37: "Colonel***",
|
||||
38: "General", 39: "General*", 40: "General**", 41: "General***",
|
||||
42: "Field Marshal", 43: "Field Marshal*", 44: "Field Marshal**", 45: "Field Marshal***",
|
||||
46: "Supreme Marshal", 47: "Supreme Marshal*", 48: "Supreme Marshal**", 49: "Supreme Marshal***",
|
||||
50: "National Force", 51: "National Force*", 52: "National Force**", 53: "National Force***",
|
||||
54: "World Class Force", 55: "World Class Force*", 56: "World Class Force**", 57: "World Class Force***",
|
||||
58: "Legendary Force", 59: "Legendary Force*", 60: "Legendary Force**", 61: "Legendary Force***",
|
||||
62: "God of War", 63: "God of War*", 64: "God of War**", 65: "God of War***",
|
||||
66: "Titan", 67: "Titan*", 68: "Titan**", 69: "Titan***",
|
||||
70: "Legends I", 71: "Legends II", 72: "Legends III", 73: "Legends IV", 74: "Legends V", 75: "Legends VI",
|
||||
76: "Legends VII", 77: "Legends VIII", 78: "Legends IX", 79: "Legends X", 80: "Legends XI", 81: "Legends XII",
|
||||
82: "Legends XIII", 83: "Legends XIV", 84: "Legends XV", 85: "Legends XVI", 86: "Legends XVII", 87: "Legends XVIII",
|
||||
88: "Legends XIX", 89: "Legends XX",
|
||||
1: 'Recruit', 2: 'Private', 3: 'Private*', 4: 'Private**', 5: 'Private***',
|
||||
6: 'Corporal', 7: 'Corporal*', 8: 'Corporal**', 9: 'Corporal***',
|
||||
10: 'Sergeant', 11: 'Sergeant*', 12: 'Sergeant**', 13: 'Sergeant***',
|
||||
14: 'Lieutenant', 15: 'Lieutenant*', 16: 'Lieutenant**', 17: 'Lieutenant***',
|
||||
18: 'Captain', 19: 'Captain*', 20: 'Captain**', 21: 'Captain***',
|
||||
22: 'Major', 23: 'Major*', 24: 'Major**', 25: 'Major***',
|
||||
26: 'Commander', 27: 'Commander*', 28: 'Commander**', 29: 'Commander***',
|
||||
30: 'Lt Colonel', 31: 'Lt Colonel*', 32: 'Lt Colonel**', 33: 'Lt Colonel***',
|
||||
34: 'Colonel', 35: 'Colonel*', 36: 'Colonel**', 37: 'Colonel***',
|
||||
38: 'General', 39: 'General*', 40: 'General**', 41: 'General***',
|
||||
42: 'Field Marshal', 43: 'Field Marshal*', 44: 'Field Marshal**', 45: 'Field Marshal***',
|
||||
46: 'Supreme Marshal', 47: 'Supreme Marshal*', 48: 'Supreme Marshal**', 49: 'Supreme Marshal***',
|
||||
50: 'National Force', 51: 'National Force*', 52: 'National Force**', 53: 'National Force***',
|
||||
54: 'World Class Force', 55: 'World Class Force*', 56: 'World Class Force**', 57: 'World Class Force***',
|
||||
58: 'Legendary Force', 59: 'Legendary Force*', 60: 'Legendary Force**', 61: 'Legendary Force***',
|
||||
62: 'God of War', 63: 'God of War*', 64: 'God of War**', 65: 'God of War***',
|
||||
66: 'Titan', 67: 'Titan*', 68: 'Titan**', 69: 'Titan***',
|
||||
70: 'Legends I', 71: 'Legends II', 72: 'Legends III', 73: 'Legends IV', 74: 'Legends V', 75: 'Legends VI',
|
||||
76: 'Legends VII', 77: 'Legends VIII', 78: 'Legends IX', 79: 'Legends X', 80: 'Legends XI', 81: 'Legends XII',
|
||||
82: 'Legends XIII', 83: 'Legends XIV', 84: 'Legends XV', 85: 'Legends XVI', 86: 'Legends XVII', 87: 'Legends XVIII',
|
||||
88: 'Legends XIX', 89: 'Legends XX',
|
||||
}
|
||||
|
||||
GROUND_RANK_POINTS: Dict[int, int] = {
|
||||
@ -186,6 +186,6 @@ GROUND_RANK_POINTS: Dict[int, int] = {
|
||||
|
||||
INDUSTRIES = Industries()
|
||||
|
||||
TERRAINS: Dict[int, str] = {0: "Standard", 1: 'Industrial', 2: 'Urban', 3: 'Suburbs', 4: 'Airport', 5: 'Plains',
|
||||
TERRAINS: Dict[int, str] = {0: 'Standard', 1: 'Industrial', 2: 'Urban', 3: 'Suburbs', 4: 'Airport', 5: 'Plains',
|
||||
6: 'Wasteland', 7: 'Mountains', 8: 'Beach', 9: 'Swamp', 10: 'Mud', 11: 'Hills',
|
||||
12: 'Jungle', 13: 'Forest', 14: 'Desert'}
|
||||
|
@ -3,15 +3,16 @@ import inspect
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import textwrap
|
||||
import time
|
||||
import traceback
|
||||
import unicodedata
|
||||
import warnings
|
||||
from base64 import b64encode
|
||||
from decimal import Decimal
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Union
|
||||
|
||||
import pytz
|
||||
import requests
|
||||
|
||||
from . import __version__, constants
|
||||
@ -25,11 +26,8 @@ __all__ = ['VERSION', 'calculate_hit', 'caught_error', 'date_from_eday', 'eday_f
|
||||
'get_air_hit_dmg_value', 'get_file', 'get_ground_hit_dmg_value', 'get_sleep_seconds', 'good_timedelta',
|
||||
'interactive_sleep', 'json', 'localize_dt', 'localize_timestamp', 'normalize_html_json', 'now',
|
||||
'process_error', 'process_warning', 'send_email', 'silent_sleep', 'slugify', 'write_file', 'write_request',
|
||||
'write_interactive_log', 'write_silent_log', 'get_final_hit_dmg', 'wait_for_lock']
|
||||
|
||||
if not sys.version_info >= (3, 6):
|
||||
raise AssertionError('This script requires Python version 3.6 and higher\n'
|
||||
'But Your version is v{}.{}.{}'.format(*sys.version_info))
|
||||
'write_interactive_log', 'write_silent_log', 'get_final_hit_dmg', 'wait_for_lock',
|
||||
'json_decode_object_hook', 'json_load', 'json_loads']
|
||||
|
||||
VERSION: str = __version__
|
||||
|
||||
@ -108,22 +106,21 @@ silent_sleep = time.sleep
|
||||
def _write_log(msg, timestamp: bool = True, should_print: bool = False):
|
||||
erep_time_now = now()
|
||||
txt = f"[{erep_time_now.strftime('%F %T')}] {msg}" if timestamp else msg
|
||||
txt = "\n".join(["\n".join(textwrap.wrap(line, 120)) for line in txt.splitlines()])
|
||||
if not os.path.isdir('log'):
|
||||
os.mkdir('log')
|
||||
with open("log/%s.log" % erep_time_now.strftime('%F'), 'a', encoding="utf-8") as f:
|
||||
f.write("%s\n" % txt)
|
||||
with open(f'log/{erep_time_now.strftime("%F")}.log', 'a', encoding='utf-8') as f:
|
||||
f.write(f'{txt}\n')
|
||||
if should_print:
|
||||
print(txt)
|
||||
|
||||
|
||||
def write_interactive_log(*args, **kwargs):
|
||||
kwargs.pop("should_print", None)
|
||||
kwargs.pop('should_print', None)
|
||||
_write_log(should_print=True, *args, **kwargs)
|
||||
|
||||
|
||||
def write_silent_log(*args, **kwargs):
|
||||
kwargs.pop("should_print", None)
|
||||
kwargs.pop('should_print', None)
|
||||
_write_log(should_print=False, *args, **kwargs)
|
||||
|
||||
|
||||
@ -131,7 +128,7 @@ def get_file(filepath: str) -> str:
|
||||
file = Path(filepath)
|
||||
if file.exists():
|
||||
if file.is_dir():
|
||||
return str(file / "new_file.txt")
|
||||
return str(file / 'new_file.txt')
|
||||
else:
|
||||
version = 1
|
||||
try:
|
||||
@ -170,17 +167,16 @@ def write_request(response: requests.Response, is_error: bool = False):
|
||||
|
||||
try:
|
||||
json.loads(html)
|
||||
ext = "json"
|
||||
ext = 'json'
|
||||
except json.decoder.JSONDecodeError:
|
||||
ext = "html"
|
||||
ext = 'html'
|
||||
|
||||
if not is_error:
|
||||
filename = f"debug/requests/{now().strftime('%F_%H-%M-%S')}_{name}.{ext}"
|
||||
write_file(filename, html)
|
||||
else:
|
||||
return {"name": f"{now().strftime('%F_%H-%M-%S')}_{name}.{ext}",
|
||||
"content": html.encode('utf-8'),
|
||||
"mimetype": "application/json" if ext == "json" else "text/html"}
|
||||
return dict(name=f"{now().strftime('%F_%H-%M-%S')}_{name}.{ext}", content=html.encode('utf-8'),
|
||||
mimetype="application/json" if ext == 'json' else "text/html")
|
||||
|
||||
|
||||
def send_email(name: str, content: List[Any], player=None, local_vars: Dict[str, Any] = None,
|
||||
@ -189,21 +185,21 @@ def send_email(name: str, content: List[Any], player=None, local_vars: Dict[str,
|
||||
local_vars = {}
|
||||
from erepublik import Citizen
|
||||
|
||||
file_content_template = "<html><head><title>{title}</title></head><body>{body}</body></html>"
|
||||
file_content_template = '<html><head><title>{title}</title></head><body>{body}</body></html>'
|
||||
if isinstance(player, Citizen) and player.r:
|
||||
resp = write_request(player.r, is_error=True)
|
||||
else:
|
||||
resp = {"name": "None.html", "mimetype": "text/html",
|
||||
"content": file_content_template.format(body="<br/>".join(content), title="Error"), }
|
||||
resp = dict(name='None.html', mimetype='text/html',
|
||||
content=file_content_template.format(body='<br/>'.join(content), title='Error'))
|
||||
|
||||
if promo:
|
||||
resp = {"name": "%s.html" % name, "mimetype": "text/html",
|
||||
"content": file_content_template.format(title="Promo", body="<br/>".join(content))}
|
||||
resp = dict(name=f"{name}.html", mimetype='text/html',
|
||||
content=file_content_template.format(title='Promo', body='<br/>'.join(content)))
|
||||
subject = f"[eBot][{now().strftime('%F %T')}] Promos: {name}"
|
||||
|
||||
elif captcha:
|
||||
resp = {"name": "%s.html" % name, "mimetype": "text/html",
|
||||
"content": file_content_template.format(title="ReCaptcha", body="<br/>".join(content))}
|
||||
resp = dict(name=f'{name}.html', mimetype='text/html',
|
||||
content=file_content_template.format(title='ReCaptcha', body='<br/>'.join(content)))
|
||||
subject = f"[eBot][{now().strftime('%F %T')}] RECAPTCHA: {name}"
|
||||
else:
|
||||
subject = f"[eBot][{now().strftime('%F %T')}] Bug trace: {name}"
|
||||
@ -213,18 +209,18 @@ def send_email(name: str, content: List[Any], player=None, local_vars: Dict[str,
|
||||
"\n".join(content)
|
||||
data = dict(send_mail=True, subject=subject, bugtrace=body)
|
||||
if promo:
|
||||
data.update({'promo': True})
|
||||
data.update(promo=True)
|
||||
elif captcha:
|
||||
data.update({'captcha': True})
|
||||
data.update(captcha=True)
|
||||
else:
|
||||
data.update({"bug": True})
|
||||
data.update(bug=True)
|
||||
|
||||
files = [('file', (resp.get("name"), resp.get("content"), resp.get("mimetype"))), ]
|
||||
filename = "log/%s.log" % now().strftime('%F')
|
||||
files = [('file', (resp.get('name'), resp.get('content'), resp.get('mimetype'))), ]
|
||||
filename = f'log/{now().strftime("%F")}.log'
|
||||
if os.path.isfile(filename):
|
||||
files.append(('file', (filename[4:], open(filename, 'rb'), "text/plain")))
|
||||
files.append(('file', (filename[4:], open(filename, 'rb'), 'text/plain')))
|
||||
if local_vars:
|
||||
if "state_thread" in local_vars:
|
||||
if 'state_thread' in local_vars:
|
||||
local_vars.pop('state_thread', None)
|
||||
|
||||
if isinstance(local_vars.get('self'), Citizen):
|
||||
@ -234,16 +230,16 @@ def send_email(name: str, content: List[Any], player=None, local_vars: Dict[str,
|
||||
if isinstance(local_vars.get('citizen'), Citizen):
|
||||
local_vars['citizen'] = repr(local_vars['citizen'])
|
||||
|
||||
from erepublik.classes import MyJSONEncoder
|
||||
files.append(('file', ("local_vars.json", json.dumps(local_vars, cls=MyJSONEncoder),
|
||||
from erepublik.classes import ErepublikJSONEncoder
|
||||
files.append(('file', ('local_vars.json', json.dumps(local_vars, cls=ErepublikJSONEncoder),
|
||||
"application/json")))
|
||||
if isinstance(player, Citizen):
|
||||
files.append(('file', ("instance.json", player.to_json(indent=True), "application/json")))
|
||||
files.append(('file', ('instance.json', player.to_json(indent=True), "application/json")))
|
||||
requests.post('https://pasts.72.lv', data=data, files=files)
|
||||
|
||||
|
||||
def normalize_html_json(js: str) -> str:
|
||||
js = re.sub(r' \'(.*?)\'', lambda a: '"%s"' % a.group(1), js)
|
||||
js = re.sub(r' \'(.*?)\'', lambda a: f'"{a.group(1)}"', js)
|
||||
js = re.sub(r'(\d\d):(\d\d):(\d\d)', r'\1\2\3', js)
|
||||
js = re.sub(r'([{\s,])(\w+)(:)(?!"})', r'\1"\2"\3', js)
|
||||
js = re.sub(r',\s*}', '}', js)
|
||||
@ -251,7 +247,7 @@ def normalize_html_json(js: str) -> str:
|
||||
|
||||
|
||||
def caught_error(e: Exception):
|
||||
process_error(str(e), "Unclassified", sys.exc_info(), interactive=False)
|
||||
process_error(str(e), 'Unclassified', sys.exc_info(), interactive=False)
|
||||
|
||||
|
||||
def process_error(log_info: str, name: str, exc_info: tuple, citizen=None, commit_id: str = None,
|
||||
@ -286,10 +282,8 @@ def process_error(log_info: str, name: str, exc_info: tuple, citizen=None, commi
|
||||
if trace:
|
||||
local_vars = trace[-1][0].f_locals
|
||||
if local_vars.get('__name__') == '__main__':
|
||||
local_vars.update({'commit_id': local_vars.get('COMMIT_ID'),
|
||||
'interactive': local_vars.get('INTERACTIVE'),
|
||||
'version': local_vars.get('__version__'),
|
||||
'config': local_vars.get('CONFIG')})
|
||||
local_vars.update(commit_id=local_vars.get('COMMIT_ID'), interactive=local_vars.get('INTERACTIVE'),
|
||||
version=local_vars.get('__version__'), config=local_vars.get('CONFIG'))
|
||||
else:
|
||||
local_vars = dict()
|
||||
send_email(name, content, citizen, local_vars=local_vars)
|
||||
@ -307,7 +301,7 @@ def process_warning(log_info: str, name: str, exc_info: tuple, citizen=None, com
|
||||
type_, value_, traceback_ = exc_info
|
||||
content = [log_info]
|
||||
if commit_id:
|
||||
content += ["Commit id: %s" % commit_id]
|
||||
content += [f'Commit id: {commit_id}']
|
||||
content += [str(value_), str(type_), ''.join(traceback.format_tb(traceback_))]
|
||||
|
||||
trace = inspect.trace()
|
||||
@ -348,7 +342,7 @@ def calculate_hit(strength: float, rang: int, tp: bool, elite: bool, ne: bool, b
|
||||
|
||||
def get_ground_hit_dmg_value(citizen_id: int, natural_enemy: bool = False, true_patriot: bool = False,
|
||||
booster: int = 0, weapon_power: int = 200) -> Decimal:
|
||||
r = requests.get(f"https://www.erepublik.com/en/main/citizen-profile-json/{citizen_id}").json()
|
||||
r = requests.get(f'https://www.erepublik.com/en/main/citizen-profile-json/{citizen_id}').json()
|
||||
rang = r['military']['militaryData']['ground']['rankNumber']
|
||||
strength = r['military']['militaryData']['ground']['strength']
|
||||
elite = r['citizenAttributes']['level'] > 100
|
||||
@ -360,7 +354,7 @@ def get_ground_hit_dmg_value(citizen_id: int, natural_enemy: bool = False, true_
|
||||
|
||||
def get_air_hit_dmg_value(citizen_id: int, natural_enemy: bool = False, true_patriot: bool = False, booster: int = 0,
|
||||
weapon_power: int = 0) -> Decimal:
|
||||
r = requests.get(f"https://www.erepublik.com/en/main/citizen-profile-json/{citizen_id}").json()
|
||||
r = requests.get(f'https://www.erepublik.com/en/main/citizen-profile-json/{citizen_id}').json()
|
||||
rang = r['military']['militaryData']['aircraft']['rankNumber']
|
||||
elite = r['citizenAttributes']['level'] > 100
|
||||
return calculate_hit(0, rang, true_patriot, elite, natural_enemy, booster, weapon_power)
|
||||
@ -380,12 +374,6 @@ def get_final_hit_dmg(base_dmg: Union[Decimal, float, str], rang: int,
|
||||
return Decimal(dmg)
|
||||
|
||||
|
||||
# def _clear_up_battle_memory(battle):
|
||||
# del battle.invader._battle, battle.defender._battle
|
||||
# for div_id, division in battle.div.items():
|
||||
# del division._battle
|
||||
|
||||
|
||||
def deprecation(message):
|
||||
warnings.warn(message, DeprecationWarning, stacklevel=2)
|
||||
|
||||
@ -409,3 +397,50 @@ def wait_for_lock(function):
|
||||
return ret
|
||||
|
||||
return wrapper
|
||||
|
||||
|
||||
def json_decode_object_hook(
|
||||
o: Union[Dict[str, Any], List[Any], int, float, str]
|
||||
) -> Union[Dict[str, Any], List[Any], int, float, str, datetime.date, datetime.datetime, datetime.timedelta]:
|
||||
""" Convert classes.ErepublikJSONEncoder datetime, date and timedelta to their python objects
|
||||
|
||||
:param o:
|
||||
:return: Union[Dict[str, Any], List[Any], int, float, str, datetime.date, datetime.datetime, datetime.timedelta]
|
||||
"""
|
||||
if o.get('__type__'):
|
||||
_type = o.get('__type__')
|
||||
if _type == 'datetime':
|
||||
dt = datetime.datetime.strptime(f"{o['date']} {o['time']}", '%Y-%m-%d %H:%M:%S')
|
||||
if o.get('tzinfo'):
|
||||
dt = pytz.timezone(o['tzinfo']).localize(dt)
|
||||
return dt
|
||||
elif _type == 'date':
|
||||
dt = datetime.datetime.strptime(f"{o['date']}", '%Y-%m-%d')
|
||||
return dt.date()
|
||||
elif _type == 'timedelta':
|
||||
return datetime.timedelta(seconds=o['total_seconds'])
|
||||
return o
|
||||
|
||||
|
||||
def json_load(f, **kwargs):
|
||||
kwargs.update(object_hook=json_decode_object_hook)
|
||||
return json.load(f, **kwargs)
|
||||
|
||||
|
||||
def json_loads(s: str, **kwargs):
|
||||
kwargs.update(object_hook=json_decode_object_hook)
|
||||
return json.loads(s, **kwargs)
|
||||
|
||||
|
||||
def b64json(obj: Union[Dict[str, Union[int, List[str]]], List[str]]):
|
||||
if isinstance(obj, list):
|
||||
return b64encode(json.dumps(obj).encode('utf-8')).decode('utf-8')
|
||||
elif isinstance(obj, (int, str)):
|
||||
return obj
|
||||
elif isinstance(obj, dict):
|
||||
for k, v in obj.items():
|
||||
obj[k] = b64json(v)
|
||||
else:
|
||||
from .classes import ErepublikException
|
||||
raise ErepublikException(f'Unhandled object type! obj is {type(obj)}')
|
||||
return b64encode(json.dumps(obj).encode('utf-8')).decode('utf-8')
|
||||
|
@ -1,19 +1,20 @@
|
||||
bump2version==1.0.1
|
||||
coverage==5.3.1
|
||||
edx-sphinx-theme==1.6.0
|
||||
coverage==5.4
|
||||
edx-sphinx-theme==1.6.1
|
||||
flake8==3.8.4
|
||||
ipython>=7.19.0
|
||||
jedi!=0.18.0
|
||||
isort==5.7.0
|
||||
pip==20.3.3
|
||||
pip==21.0
|
||||
pre-commit==2.9.3
|
||||
pur==5.3.0
|
||||
PyInstaller==4.1
|
||||
PyInstaller==4.2
|
||||
PySocks==1.7.1
|
||||
pytest==6.2.1
|
||||
pytest==6.2.2
|
||||
pytz>=2020.5
|
||||
requests>=2.25.1
|
||||
responses==0.12.1
|
||||
setuptools==51.1.2
|
||||
setuptools==52.0.0
|
||||
Sphinx==3.4.3
|
||||
twine==3.3.0
|
||||
wheel==0.36.2
|
||||
|
@ -1,5 +1,5 @@
|
||||
[bumpversion]
|
||||
current_version = 0.23.4.9
|
||||
current_version = 0.24.0.1
|
||||
commit = True
|
||||
tag = True
|
||||
parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)\.?(?P<dev>\d+)?
|
||||
|
Reference in New Issue
Block a user