def __init__(self):
self.br = PathBrowser()
self.logged_in = False
self.db = Db()
# always reset the db on a new session
self.db.clear()
self.db.init()
class PathSession(object):
def __init__(self):
self.br = PathBrowser()
self.logged_in = False
self.db = Db()
# always reset the db on a new session
self.db.clear()
self.db.init()
def login_normal(self, username, password):
self.br.open(settings["URL_LOGIN"])
self.br.select_form(nr=0)
self.br.form["login_email"] = username
self.br.form["login_password"] = password
self.br.submit()
self.check_logged_in()
def login_session_id(self, session_id):
if len(session_id) != 32:
raise InvalidSessionIDException(
"Session ID needs to be 32 characters")
self.br.open(settings["URL_LOGIN"])
cj = mechanize.LWPCookieJar()
self.br.set_cookiejar(cj)
cookie = mechanize.Cookie(
version=0,
name="PHPSESSID",
value=session_id,
port=None,
port_specified=False,
domain="www.pathofexile.com",
domain_specified=False,
domain_initial_dot=False,
path="/",
path_specified=True,
secure=False,
expires=None,
discard=True,
comment=None,
comment_url=None,
rest=None,
)
cj.set_cookie(cookie)
self.check_logged_in()
def check_logged_in(self):
self.br.open(settings["URL_ACCOUNT"])
if self.br.geturl() != settings["URL_ACCOUNT"]:
self.logged_in = False
raise InvalidLoginException("Could not log in, bad credentials")
else:
self.logged_in = True
def logged_in(func):
from functools import wraps
@wraps(func)
def wrapper(self, *args, **kwargs):
if self.logged_in:
return func(self, *args, **kwargs)
else:
raise NotLoggedInException("Not logged in")
return wrapper
@logged_in
def get_stash(self, league="standard"):
result = True
tab_index = 0
# TODO: temporary override of max number of stash tabs for testing
while result is not None and tab_index < 1:
delay = 0 if tab_index % 5 == 0 else 0
# TODO: Threading
stash_tab = self.get_stash_tab(league, tab_index, delay)
items = []
tab = get_or_create(
self.db.session,
StashTab,
league=stash_tab["items"][0]["league"],
name=stash_tab["items"][0]["inventoryId"],
tab_index=tab_index)
for item in stash_tab["items"]:
items.append(Item(
stash_tab=tab,
name=item["name"],
icon=item["icon"],
type_line=item["typeLine"],
verified=item["verified"],
identified=item["identified"],
support=item["support"],
frame_type=item["frameType"],
w=item["w"],
h=item["h"],
x=item["x"],
y=item["y"]))
self.db.session.add(tab)
self.db.session.add_all(items)
self.db.session.commit()
tab_index += 1
@logged_in
def get_stash_tab(self, league="standard", tab=0, delay=0):
time.sleep(delay)
url = settings["URL_STASH"] + ("?league=%s&tabIndex=%s" % (league, tab))
response = self.br.open(url)
content = response.read()
try:
data = json.loads(content)
if data is False:
return None
# I don't think this ever happens
except ValueError:
raise DataNotFoundException("Invalid data returned")
logger.info(data)
return data
