def savechars(self):
'''
Saves current characters to file
'''
console = ConsoleEngine()
t = console.logtask("Saving character sessions to file...")
with self.filelock:
# Locks cannot be pickled, so prepare a lockless (shallow) copy
charcopy = {}
for x in self.getchars():
char = self.chars[x]
# Acquire all char locks while we rig up conn_session
char.lock.acquire()
charcopy[char.name] = copy.copy(char)
charcopy[char.name].lock = None
# Hack conn_session into existence
charcopy[char.name].conn_session.__attrs__.append('robin')
if (os.path.exists(Config.charsave)):
# Make a backup of previous save
shutil.copy(Config.charsave, Config.charsave + ".bak")
f = open(Config.charsave, 'w')
pickle.dump(charcopy, f)
f.close()
# Undo our conn_session rigging so that it can be used properly again
for x in self.getchars():
self.chars[x].conn_session.__attrs__.remove('robin')
self.chars[x].lock.release()
console.logstatus(t, "[OK]")
def update(self):
with self.lock:
conn = ROBConn()
console = ConsoleEngine()
t = console.logtask("Updating stats for [{}]".format(self.name))
self.updating = True
# When doing arbitrary updates, emulate navbar press (i.e. send session)
soup = conn.geturl('mypage', self.conn_session, send_session=self.session, send_viewer=self.viewer)
soup = self.handlewelcome(soup)
# Hmm. We only need to do this once, really.
self.name = soup.find("div", "area_name").get_text().strip()
self.conn_session.robin['name'] = self.name
# Stats
rawcards = soup.find("a", text="Cards").parent.find_next_sibling().get_text().split("/")
self.cards = int(rawcards[0])
self.maxcards = int(rawcards[1])
self.rupies = soup.find('span', text=re.compile("Rupies:")).parent.find_next_sibling().get_text()
self.rupies = int(self.rupies)
rawstam = soup.find(text=re.compile("STAMINA:")).parent.span.get_text().strip().split("/")
self.stamina = int(rawstam[0])
self.maxstamina = int(rawstam[1])
# Level and attribute points
rawlevel = soup.find(text=re.compile("LVL:")).find_parent().span.get_text().strip()
rawlevel = re.match("([0-9]*)(\(([0-9]*)\))?", rawlevel)
self.level = int(rawlevel.group(1))
if (rawlevel.group(3) is not None):
self.points = int(rawlevel.group(3))
else:
self.points = 0
# Scrape Inbox
inboxlinks = soup.find(text=re.compile(r'Inbox(?! \()')).parent.find_next_sibling().find_all("a")
self.inbox = [(re.match("http://bahamut-n.cygames.jp/bahamut_n/(.*)", x['href']).group(1), x.get_text()) for x in inboxlinks]
self.updating = False
self.lastupdatetime = datetime.datetime.now()
# Set next update anywhere between 10mins and 90mins
self.nextupdate = datetime.timedelta(seconds=(random.randint(10, 90) * 60))
console.logstatus(t, "[OK]")
def addchar(self, option, agent):
if (option.lower() == "adb"):
console = ConsoleEngine()
t = console.logtask("Character add from ADB requested... (Ctrl-C cancels)")
try :
f = os.popen("adb -d logcat -d")
d = f.read()
c = re.findall(r"session_id=[a-z0-9]*&viewer_id=[0-9]*", d)
c.reverse()
except KeyboardInterrupt:
console.logstatus(t, "[CANCELLED]")
return "CANCELLED"
if (len(c) == 0):
console.logstatus(t, "[FAILED]")
return "FAILED"
else:
rawsess = c[0]
console.logstatus(t, "[OK]")
else:
rawsess = option
rawsess = rawsess.split("&")
session = rawsess[0].split("=")[1].strip()
viewer = rawsess[1].split("=")[1].strip()
char = ROBChar(session, viewer, agent)
# If under maintenance
if self.conn.maintenance:
return "MAINTENANCE"
else:
self.chars[char.name] = char
return char.name
def __init__(self, session, viewer, agent):
'''
Initializes cookies, auto-refresh
'''
conn = ROBConn()
console = ConsoleEngine()
self.conn_session = conn.getnewsession(agent)
self.agent = agent
self.session = session
self.viewer = viewer
self.name = "<new>"
# Ride on conn_session with some of our own data
self.conn_session.robin = {}
self.conn_session.robin['name'] = self.name
self.conn_session.robin['expired'] = False
self.conn_session.robin['viewer'] = self.viewer
# Lock whenever modifications are being made
self.lock = threading.Lock()
# 1. Call ROB homepage to initialize cookies
t = console.logtask("Initializing cookies for new character...")
conn.geturl('', self.conn_session, send_session=self.session, send_viewer=self.viewer)
self.logontime = datetime.datetime.now()
if conn.maintenance:
console.logstatus(t, "[MAINTENANCE]")
return
else:
console.logstatus(t, "[OK]")
# 2. Scrape 'My Page', check for welcome event
t = console.logtask("Retrieving character stats...")
self.update()
console.logstatus(t, "[OK]")
console.log("== New character: [{}] ({})".format(self.name, self.agent))
def loadchars(self):
'''
Loads characters from file
Characters loaded are automatically scheduled for an update between 0.5 and 10mins later
'''
console = ConsoleEngine()
y = console.logtask("Loading character sessions from file...")
t = MenuStr()
with self.filelock:
try:
f = open(Config.charsave, 'r')
tempchars = pickle.load(f)
f.close()
except IOError:
console.logstatus(y, "[FAILED]")
console.log("File '{}' not available for reading.".format(Config.charsave))
return
except:
console.logstatus(y, "[FAILED]")
console.log("Error reading '{}'".format(Config.charsave))
return
for x in tempchars.keys():
tchar = tempchars[x]
# Check last update time - we don't want anything too old
currtime = datetime.datetime.now()
sinceupdate = currtime - tchar.lastupdatetime
maxage = Config.maxloadage
if (maxage > 150):
maxage = 150
smaxage = maxage * 60
if (sinceupdate.seconds > smaxage):
# Too old
console.log("Load: Skipped [{}] - Session too stale".format(tchar.name))
t.add("Skipping [{}]: Session too stale.".format(tchar.name))
continue
# Check if same as existing character
if (tchar.name in self.chars):
# If so, see which has a fresher logon
tsl = currtime - tchar.logontime
sl = currtime - self.chars[tchar.name].logontime
if (tsl < sl):
console.log("Load: Overrode [{}] - Fresher logon".format(tchar.name))
t.add("Overriding [{}]: Fresher logon.".format(tchar.name))
else:
console.log("Load: Skipped [{}] - Not fresher than existing logon".format(tchar.name))
t.add("Skipping [{}]: Not fresher than existing logon.".format(tchar.name))
continue
else:
# New character
console.log("Load: Added [{}]".format(tchar.name))
t.add("Adding [{}]".format(tchar.name))
# If we haven't continued by now, it's a valid character
self.chars[tchar.name] = tchar
self.chars[tchar.name].lock = threading.Lock()
# Done. Let the other threads have at it
self.chars[tchar.name].nextupdate = datetime.timedelta(seconds=(random.randint(30, 600)))
console.logstatus(y, "[OK]")
return t.get()