Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-08-08 14:32:23 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Read/Write Cookie DB file when creating and stopping browser instance.

Browse source
This commit is contained in:
Adam Miller 2016-07-22 00:22:28 +00:00
parent 127002b77d
commit 1cb6653fab
3 changed files with 31 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

28
brozzler/browser.py
View file

@ -128,11 +128,19 @@ class Browser:
def __exit__(self, *args):
self.stop()
def start(self, proxy=None):
def start(self, proxy=None, cookieDb=None):
if not self._chrome_instance:
# these can raise exceptions
self.chrome_port = self._find_available_port()
self._work_dir = tempfile.TemporaryDirectory()
if cookieDb is not None:
cookieLocation = os.sep.join([self._work_dir.name, "chrome-user-data","Default","Cookies"])
try:
with open(cookieLocation,'w') as cookieFile:
cookieFile.write(cookieDb)
except EnvironmentError:
self.logger.error("exception writing cookie file at: %s", cookieLocation, exc_info=True)
self._chrome_instance = Chrome(
port=self.chrome_port, executable=self.chrome_exe,
user_home_dir=self._work_dir.name,
@ -161,6 +169,24 @@ class Browser:
except:
self.logger.error("problem stopping", exc_info=True)
def read_cookie_db(self):
cookieLocation = os.sep.join([self._work_dir.name, "chrome-user-data","Default","Cookies"])
try:
with sqlite3.connect(cookieLocation) as conn:
cur = conn.cursor()
cur.execute("UPDATE cookies SET persistent = 1")
except sqlite3.Error:
self.logger.error("exception updating cookie DB", exc_info=True)
cookieDb=None
try:
with open(cookieLocation, "rb") as cookieFile:
cookieDb=bytearray(cookieFile.read())
except EnvironmentError:
self.logger.error("exception reading from cookie DB file at: %s", cookieLocation, exc_info=True)
return cookieDb
def _find_available_port(self):
port_available = False
port = self.chrome_port