MarketBot/buy_bot.py

from rival_regions_wrapper.middleware import LocalAuthentication
from rival_regions_wrapper import util
from rival_regions_wrapper.wrapper.abstract_wrapper import AbstractWrapper
import re
from bs4 import BeautifulSoup


authentication = LocalAuthentication(True)
authentication.set_credentials("mail@gmail.com", "password", "google")


class CurrentPrice(AbstractWrapper):
    def info(self, resource):
        """Get profile"""
        if isinstance(resource, str) and resource in util.ITEM_KEYS:
            resource = util.ITEM_KEYS[resource]
        path = "storage/listed/{}".format(resource)
        response = self.middleware.get(path)
        soup = BeautifulSoup(response, "html.parser")
        offers_tree = soup.find_all(class_="list_link")
        test = 0
        prijs = 0
        for offer_tree in offers_tree:
            if test == 0:
                test = 1
                prijs = int(
                    float(offer_tree.select(".list_level")[1]["rat"]) * 100
                )
        return prijs


class Amount(AbstractWrapper):
    def info(self, resource):
        """Get profile"""
        if isinstance(resource, str) and resource in util.ITEM_KEYS:
            resource = util.ITEM_KEYS[resource]
        path = "storage/listed/{}".format(resource)
        response = self.middleware.get(path)
        soup = BeautifulSoup(response, "html.parser")
        offers_tree = soup.find_all(class_="list_link")
        test = 0
        for offer_tree in offers_tree:
            if test == 0:
                test = 1
                amount = int(
                    offer_tree.select_one(".list_level.imp.small")["rat"]
                )
        return amount


class Profile(AbstractWrapper):
    def info(self, resource):
        """Get profile"""
        if isinstance(resource, str) and resource in util.ITEM_KEYS:
            resource = util.ITEM_KEYS[resource]
        path = "storage/listed/{}".format(resource)
        response = self.middleware.get(path)
        soup = BeautifulSoup(response, "html.parser")
        offers_tree = soup.find_all(class_="list_link")
        test = 0
        profiel = 0
        for offer_tree in offers_tree:
            if test == 0:
                test = 1
                profiel = int(
                    re.sub(
                        r"^.*\/",
                        "",
                        offer_tree.select_one(".results_date")["action"],
                    )
                )
        return profiel


class BuyMarket(AbstractWrapper):
    def info(self, resource, price):
        current_price = CurrentPrice(authentication).info(resource) / 100
        if current_price < price:
            profiel = Profile(authentication).info(resource)
            amount = Amount(authentication).info(resource)
            item = util.ITEM_KEYS[resource]
            self.middleware.post(
                "storage/buy/{}/{}/{}/{}".format(
                    item, profiel, amount, current_price
                )
            )
        return True