From 13cacaed6b35fac1c67bdaeadfee2f537fd7afc5 Mon Sep 17 00:00:00 2001
From: Raphael Roberts <raphael.roberts48@gmail.com>
Date: Mon, 6 Jan 2020 01:40:17 -0600
Subject: [PATCH] Added setup.py and some other improvements

---
 .gitignore               |  1 +
 requirements.txt         |  1 +
 rlbr_browser/__init__.py | 14 ++++++++++++++
 rlbr_browser/browser.py  |  3 +++
 setup.py                 | 14 ++++++++++++++
 5 files changed, 33 insertions(+)
 create mode 100644 setup.py

diff --git a/.gitignore b/.gitignore
index aad875c..4bf0fd1 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 /.dir-locals.el
 __pycache__
+*.egg-info/
diff --git a/requirements.txt b/requirements.txt
index 2191931..fb2d0e9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ websockets==6.0
 pyppeteer
 requests
 lxml
+appdirs
diff --git a/rlbr_browser/__init__.py b/rlbr_browser/__init__.py
index e69de29..9e20267 100644
--- a/rlbr_browser/__init__.py
+++ b/rlbr_browser/__init__.py
@@ -0,0 +1,14 @@
+import os
+
+import appdirs
+
+app_dirs = appdirs.AppDirs("rlbr_browser")
+config_dir = app_dirs.user_config_dir
+cache_dir = app_dirs.user_cache_dir
+log_dir = app_dirs.user_log_dir
+state_dir = app_dirs.user_state_dir
+
+
+for dir in (config_dir, cache_dir, log_dir, state_dir):
+    if not os.path.exists(dir):
+        os.makedirs(dir)
diff --git a/rlbr_browser/browser.py b/rlbr_browser/browser.py
index 54a99c4..43757d5 100644
--- a/rlbr_browser/browser.py
+++ b/rlbr_browser/browser.py
@@ -46,6 +46,9 @@ class Tab:
         return self
 
     def __exit__(self, *args):
+        self.close()
+
+    def close(self):
         run(self.page_handle.close())
 
     def get_source(self):
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..cc1cb22
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,14 @@
+from setuptools import setup, find_packages
+
+with open("requirements.txt") as file:
+    INSTALL_REQUIRES = file.read().rstrip().split("\n")
+
+setup(
+    name="rlbr_browser",
+    version="1.0",
+    description="Convienience wrapper around pyppeteer",
+    install_requires=INSTALL_REQUIRES,
+    author="Raphael Roberts",
+    author_email="raphael.roberts48@gmail.com",
+    packages=find_packages(),
+)