async def idle(self): # work cooldown time work_c = 600 war_c, is_traveling, _ = await self.check_overview() await self.do_perks_upgrade() if is_traveling: await self.sleep(600) return await self.do_storage_supply() if war_c == 0: await self.do_military_training() work_c = await self.do_work() war_c, _, _ = await self.check_overview() perk_c = await self.calculate_perks_time() gold, money = Status.check_str_money(await self.get_soup()) LOG.info(f"Your property: {money} $ {gold} G") await self.sleep(min(perk_c, war_c, work_c))
async def save_cookies(self): cookies = await self.browser.cookies('https://rivalregions.com/', 'https://www.facebook.com/', 'https://accounts.google.com/', 'https://oauth.vk.com/') pickle.dump(cookies, open("{}.pkl".format(self.profile), "wb")) LOG.info("Cookies saved!")
def decision(rushee, decision_type): rushee = Rushee.query.filter(Rushee.id == rushee)[0] rushee.decided = decision_type db.session.commit() LOG.info("Wrote %s to %s list" % (rushee.name, decision_type)) flash('Successfully added %s to %s list' % (rushee.name, decision_type)) return redirect(request.referrer)
async def load_cookies(self): try: cookies = pickle.load(open("{}.pkl".format(self.profile), "rb")) for cookie in cookies: await self.browser.set_cookie(cookie) LOG.info("Cookies loaded!") except Exception: LOG.info('First Login') if self.headless: LOG.error('Please do not use headless mode!')
async def do_military_training(self) -> int: try: await self.click(War.selector(), 1) await self.click(War.military_training_selector(), 1) await self.click(War.send_ok_selector(), 1) await self.click(utils.close_selector(), 2) LOG.info("Military training completed") except Exception as err: LOG.debug(err) LOG.error("Military training failed") return 3598
async def main(): parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter) parser.add_argument("-l", "--login_method", help="登入選項: 'GOOGLE'、'FB'、'VK'", dest='login_method') parser.add_argument( "-u", "--use_to_upgrade", help='''升級道具: 'RRCash'、'GOLD'、'GOLD1' 預設使用 RRCash 1. 'GOLD' 當GOLD小於4320將會改用RRCash 2. 'GOLD1' 同1,並且對於upgrade_strategy中的最小值會使用RRCash升級 例如: "2:1:1" STR將會使用GOLD, EDU, END將會使用RRCash 也就是窮人的選擇''', default='RRCash', dest='use_to_upgrade' ) parser.add_argument("-p", "--profile", help="帳戶profile: 預設為'default', 修改可更換帳戶", default='default', dest='profile') parser.add_argument("--poor", help="你是窮人, 你買不起高級會員, 你必須手動挖礦、軍演, 可憐哪 我來幫你", action="store_true") parser.add_argument("--headless", help="確定使用者有登入成功後可開啟 將瀏覽器GUI關閉節省資源", action="store_true", dest='headless') parser.add_argument("--proxy", help="正確格式如下: socks5://localhost:1080, https://localhost:1080", dest='proxy') parser.add_argument( "--upgrade_strategy", help="三圍100後將按照時間比例來升級 default '2:1:1'", default='2:1:1', dest="upgrade_strategy" ) parser.add_argument("--debug", help="開啟Debug mod", action="store_true") args = parser.parse_args() if (args.login_method is None): parser.print_help() else: initLog(args.profile, DEBUG=args.debug) while (True): try: r = None if args.poor is True: from app.poorbot import PoorBot r = await PoorBot(**vars(args)) else: from app.rrbot import RRBot r = await RRBot(**vars(args)) await r.start() break except Exception as err: LOG.debug('Bot detect error: {}'.format(err)) # Due to "browser.close" sometime not working, use kill child processes instead. kill_child_processes(os.getpid()) if r: await r.quit() LOG.info('Restarting...')
def add(): form = RusheeForm() if form.validate_on_submit(): LOG.info(form.pic.data) rushee = Rushee(form.name.data, form.computing_id.data, form.year.data, form.dorm.data, form.pic.data) db.session.add(rushee) db.session.commit() flash('Thanks for checking in') LOG.info("Successfully logged: %s (%s)" % (form.name.data, form.computing_id.data)) return redirect(url_for('add')) else: flash_errors(form) return render_template('add.html', form=form)
async def idle(self): await self.refresh() await self.check_login() await self.do_perks_upgrade() await self.do_storage_supply() _, _, (energy, energy_cooldown_time) = await self.check_overview() if (energy >= 20) and energy_cooldown_time == 0: LOG.warn( "There is no gold in your area or automatic working expires") time = await self.calculate_perks_time() if time: gold, money = Status.check_str_money(await self.get_soup()) LOG.info(f"Your property: {money} $ {gold} G") await self.sleep(3600 if time >= 3600 else time)
async def do_work(self) -> int: await self.click(Work.selector(), 3) await self.browser.wait_for(".imp.yellow.tip") soup = await self.get_soup() energy, energy_cooldown_time = Status.check_energy(soup) gold = Work.check_region_gold(soup) if not Work.can_work(soup): LOG.info("Working is not possible") return 600 if gold > 0 and energy >= 10: try: await self.click(Work.work_selector(), 3) await self.click(utils.close_selector(), 3) LOG.info("Work is completed {} energys use to work".format(energy)) except Exception as err: LOG.debug(err) LOG.error("Can not work, maybe the factory owner doesn't have enough money?") return 600 elif gold > 0 and energy_cooldown_time == 0 and int(self.perks['END'][0]) >= 50: await self.click(Status.energy_bar_selector(), 3) else: if int(self.perks['END'][0]) < 50: return 600 elif gold == 0: LOG.info("Region lack of gold") return 600 elif energy >= 10 or energy_cooldown_time == 0: LOG.error("Some error occurred in work") return 600 return energy_cooldown_time return await self.do_work()
async def __init__(self, login_method="GOOGLE", use_to_upgrade="RRCash", profile="default", upgrade_strategy="2:1:1", proxy=None, headless=None, poor=None, debug=False): self.profile = profile self.uri = "https://rivalregions.com" self.headless = headless if proxy: proxy = "--proxy-server={}".format(proxy) self.browser = await Browser(headless, proxy, dumpio=True) self.browser.set_default_navigation_timeout(30000) self.login_method = login_method if use_to_upgrade not in ['RRCash', 'GOLD', 'GOLD1']: self.use_to_upgrade = 'RRCash' else: self.use_to_upgrade = use_to_upgrade LOG.info(f"Will use {self.use_to_upgrade} upgrade perks.") try: self.upgrade_strategy = [ int(x) for x in upgrade_strategy.split(':') ] assert len(self.upgrade_strategy) == 3 except Exception: LOG.error("--upgrade_strategy invalid") self.perks = {'strategy': self.upgrade_strategy} await self.load_cookies() await self.browser.goto(self.uri, waitUntil='networkidle0') await self.check_login()
async def login(self): LOG.debug("Login") await self.browser.goto(self.uri, waitUntil='networkidle0') selectors = { 'FB': '.sa_link[href*="facebook.com"]', 'GOOGLE': '.sa_link[href*="google.com"]', 'VK': '.sa_link[href*="vk.com"]' } try: LOG.debug("Try login") await self.browser.click(selectors[self.login_method]) await self.browser.wait_for_response('https://rivalregions.com/', timeout=240000) await self.browser.wait_for('#chat input[name=name]', timeout=10000) name = await self.browser.query_selector_eval( '#chat input[name=name]', 'node => node.value') LOG.info("Login success {}".format(name)) await self.save_cookies() except Exception as err: LOG.debug(err) await self.check_login()
async def upgrade(self, perk): selector = None gold, _ = Status.check_money(await self.get_soup()) minimal_strategy = [ i for i, v in enumerate(self.upgrade_strategy) if v == min(self.upgrade_strategy) ] if len(minimal_strategy) == 3: minimal_strategy = [] if self.upgrade_strategy == 'RRCash' or gold <= 4320: upgrade_selector = "#perk_target_4 > div[url='1'] > div > div" elif self.use_to_upgrade == "GOLD" or (self.use_to_upgrade == "GOLD1" and perk.value not in minimal_strategy): upgrade_selector = "#perk_target_4 > div[url='2'] > div > div" else: upgrade_selector = "#perk_target_4 > div[url='1'] > div > div" if perk == Perks.STR: selector = ".perk_item[perk='1']" elif perk == Perks.EDU: selector = ".perk_item[perk='2']" elif perk == Perks.END: selector = ".perk_item[perk='3']" else: raise Exception("Perk not found") await self.click(selector, 3) await self.click(upgrade_selector, 3) LOG.info("Upgrading {}: {} -> {}".format(perk.name, self.perks[perk.name][0], self.perks[perk.name][0] + 1))
async def do_storage_supply(self): LOG.debug("Do storage supply") await self.click(Storage.selector(), wait_for="span[action='leader/storage']") soup = BeautifulSoup((await self.browser.content()), "html5lib") # Produce energy drink if int( soup.find("span", { "urlbar": str(Storage.Energydrink.value) }).text.replace('.', '')) <= 3600: await self.click(Storage.selector(Storage.Energydrink.value), 3) gold, _ = Status.check_money(await self.get_soup()) # 6 hours ~ 24 hours amount = random.randrange(10800, 43200, 1000) if gold < amount / 10: amount = gold * 10 if amount > 300: await self.type('.storage_produce_ammount', str(amount)) await self.click('.storage_produce_button', 3) LOG.info('Produced: energy drink {} pcs.'.format(amount)) need_products = [(Storage.Bombers, 10000), (Storage.Moontanks, 10000), (Storage.Spacestations, 1280)] # Buy products for pair in need_products: product, number = pair if int( soup.find("span", { "urlbar": str(product.value) }).text.replace('.', '')) < number: await self.buy_product(product, number)
async def buy_product(self, product, amount): """ Must be #storage page : product (Storage): Storage.Bombers、..... : amount (int) """ await self.click(Storage.selector(product.value), 3) price, num = Storage.check_product_price(await self.get_soup()) _, money = Status.check_money(await self.get_soup()) # Reserve 1 billion(1kkk) for upgrade perk money -= 1000000000 if money < price: return amount = amount if num >= amount else num if price * amount > money: amount = money // price await self.type('.storage_buy_input', str(amount)) await self.click('.storage_buy_button', 3) LOG.info("Market purchase: {}, {} pcs, total {}.".format( product.name, amount, price * amount))
def add_to_logs(message, user): """ Method to important info messages to different logs. """ print(message, "User: {}".format(user)) LOG.info(message + " User: {}".format(user))
async def start(self): LOG.info('Bot start') while (True): await self.idle() await self.sleep(5)
async def sleep(self, sec=randint(10, 30)): if (sec >= 120): LOG.info("Wait for {:.1f} minutes".format(sec / 60.0)) await asyncio.sleep(sec)