chore: update project infrastructure and dependencies

- Update run.sh with comprehensive Python 3.12 and Poetry setup
- Update dependencies in poetry.lock and pyproject.toml
- Add LICENSE file modifications

This commit improves the project setup process and ensures consistent
Python 3.12 environment across different systems.

Author: gr8monk3ys

.env.example

@@ -1,4 +1,4 @@
-AGENTQL_API_KEY=""
-OPENAI_API_KEY=""
-LEETCODE_USERNAME=""
+AGENTQL_API_KEY=""
+OPENAI_API_KEY=""
+LEETCODE_USERNAME=""
 LEETCODE_PASSWORD=""

.gitignore

@@ -1,2 +1,3 @@
-parameters.py
-.env
+parameters.py
+.env
+leetcode_login.json

LICENSE

@@ -1,20 +1,20 @@
-[tool.poetry]
-name = "leetcode-solver"
-version = "0.1.0"
-description = "A Leetcode bot"
-authors = ["Lorenzo Scaturchio"]
-license = "GNU General Public License"
-readme = "README.md"
-
-[tool.poetry.dependencies]
-python = ">=3.12,<=3.13"
-selenium = "^4.27.1"
-beautifulsoup4 = "^4.12.3"
-openai = "^1.57.0"
-python-dotenv = "^1.0.1"
-playwright = "^1.41.2"
-agentql = "^1.6.1"
-
-[build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+[tool.poetry]
+name = "leetcode-solver"
+version = "0.1.0"
+description = "A Leetcode bot"
+authors = ["Lorenzo Scaturchio"]
+license = "GNU General Public License"
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = ">=3.12,<=3.13"
+selenium = "^4.27.1"
+beautifulsoup4 = "^4.12.3"
+openai = "^1.57.0"
+python-dotenv = "^1.0.1"
+playwright = "^1.41.2"
+agentql = "^1.6.1"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

poetry.lock

@@ -10,12 +10,16 @@
 from utils.select_random import select_random
 from utils.solve_problem import solve_problem
 
+load_dotenv()
+os.environ["AGENTQL_API_KEY"] = os.getenv('AGENTQL_API_KEY')
+INITIAL_URL = "https://leetcode.com/accounts/login/"
+
 # Setup logging
 logging.basicConfig(
     level=logging.DEBUG,
     format='%(asctime)s - %(levelname)s - %(message)s',
     handlers=[
-        logging.FileHandler('leetcode_solver.log', encoding='utf-8'),
+        logging.FileHandler('logs/leetcode_solver.log', encoding='utf-8'),
         logging.StreamHandler()
     ]
 )
@@ -29,10 +33,21 @@ def main():
             page = agentql.wrap(browser.new_page())
             time_start = time.time()
 
-            try:
-                # Login
-                if not login(page):
+            state_path = os.path.join(os.getcwd(), "leetcode_login.json")
+            
+            # Login
+            if not os.path.exists(state_path):
+                logging.info("No login state found. Logging in...")
+                if not login(page, INITIAL_URL, browser):
                     raise Exception("Failed to login")
+                logging.info("Successfully logged in to Leetcode")
+            
+            try:
+                # Create new context with saved login state
+                context = browser.new_context(storage_state=state_path)
+                page = agentql.wrap(context.new_page())
+                page.goto("https://leetcode.com")
+                page.wait_for_load_state("networkidle")
 
                 # Select random problem
                 if not select_random(page):
@@ -46,13 +61,17 @@ def main():
 
                 time_elapsed = time.time() - time_start
                 logging.info(f"Total time elapsed: {time_elapsed:.2f} seconds")
-            
-            finally:
-                browser.close()
+                
+            except Exception as e:
+                logging.error(f"Error during execution: {str(e)}")
+                if os.path.exists(state_path):
+                    logging.info("Removing invalid login state file")
+                    os.remove(state_path)
+                raise e
 
     except Exception as e:
         logging.error(f"Error encountered: {str(e)}")
-        raise
+        raise e
 
 if __name__ == '__main__':
     main()

pyproject.toml

@@ -1,26 +1,26 @@
-echo "Running the bot..."
-
-# Update package list and install prerequisites
-sudo apt update
-sudo apt install -y software-properties-common
-
-# Add deadsnakes PPA and install Python 3.12
-sudo add-apt-repository ppa:deadsnakes/ppa
-sudo apt update
-sudo apt install -y python3.12 python3.12-venv python3.12-dev
-
-# Ensure pip is installed for Python 3.12
-curl -sS https://bootstrap.pypa.io/get-pip.py | sudo python3.12
-
-# Install pipx
-python3.12 -m pip install --user pipx
-python3.12 -m pipx ensurepath
-
-# Install Poetry using pipx
-pipx install poetry
-
-# Install project dependencies using Poetry
-poetry install
-
-# Run the bot
+echo "Running the bot..."
+
+# Update package list and install prerequisites
+sudo apt update
+sudo apt install -y software-properties-common
+
+# Add deadsnakes PPA and install Python 3.12
+sudo add-apt-repository ppa:deadsnakes/ppa
+sudo apt update
+sudo apt install -y python3.12 python3.12-venv python3.12-dev
+
+# Ensure pip is installed for Python 3.12
+curl -sS https://bootstrap.pypa.io/get-pip.py | sudo python3.12
+
+# Install pipx
+python3.12 -m pip install --user pipx
+python3.12 -m pipx ensurepath
+
+# Install Poetry using pipx
+pipx install poetry
+
+# Install project dependencies using Poetry
+poetry install
+
+# Run the bot
 poetry run python bot.py