Browse Source

Move credentials to env

JoostSijm 3 years ago
parent
commit
2f625c3845
4 changed files with 67 additions and 9 deletions
  1. 1 0
      .gitignore
  2. 30 5
      buy_bot.py
  3. 7 0
      example.env
  4. 29 4
      sell_bot.py

+ 1 - 0
.gitignore

@@ -0,0 +1 @@
+.env

+ 30 - 5
buy_bot.py

@@ -1,12 +1,37 @@
-from rival_regions_wrapper.middleware import LocalAuthentication
-from rival_regions_wrapper import util
-from rival_regions_wrapper.wrapper.abstract_wrapper import AbstractWrapper
+"""Buy bot"""
+
 import re
+import os
+
 from bs4 import BeautifulSoup
+from rival_regions_wrapper.middleware import LocalAuthentication
+from rival_regions_wrapper.wrapper.abstract_wrapper import AbstractWrapper
+from rival_regions_wrapper import util
+
+
+class MissingAuthenticationError(Exception):
+    """Error for missing authentication"""
+
+
+RIVAL_REGIONS_USERNAME = os.environ.get("RIVAL_REGIONS_USERNAME", None)
+RIVAL_REGIONS_PASSWORD = os.environ.get("RIVAL_REGIONS_PASSWORD", None)
+RIVAL_REGIONS_LOGIN_METHOD = os.environ.get("RIVAL_REGIONS_LOGIN_METHOD", None)
+CAPTCHA_KEY = os.environ.get("CAPTCHA_KEY", None)
 
+if None in (
+    RIVAL_REGIONS_USERNAME,
+    RIVAL_REGIONS_PASSWORD,
+    RIVAL_REGIONS_LOGIN_METHOD,
+):
+    raise MissingAuthenticationError(
+        "Load the following variables in your user environment: "
+        "RIVAL_REGIONS_USERNAME, RIVAL_REGIONS_PASSWORD, RIVAL_REGIONS_LOGIN_METHOD"
+    )
 
-authentication = LocalAuthentication(True)
-authentication.set_credentials("mail@gmail.com", "password", "google")
+authentication = LocalAuthentication(False, CAPTCHA_KEY)
+authentication.set_credentials(
+    RIVAL_REGIONS_USERNAME, RIVAL_REGIONS_PASSWORD, RIVAL_REGIONS_LOGIN_METHOD
+)
 
 
 class CurrentPrice(AbstractWrapper):

+ 7 - 0
example.env

@@ -0,0 +1,7 @@
+# Captcha
+CAPTCHA_KEY=PLACEHOLDER
+
+# Rival Regions
+RIVAL_REGIONS_USERNAME=PLACEHOLDER
+RIVAL_REGIONS_PASSWORD=PLACEHOLDER
+RIVAL_REGIONS_LOGIN_METHOD=PLACEHOLDER

+ 29 - 4
sell_bot.py

@@ -1,12 +1,37 @@
+"""Sell bot"""
+
+import re
+import os
+
+from bs4 import BeautifulSoup
 from rival_regions_wrapper.middleware import LocalAuthentication
 from rival_regions_wrapper import util
 from rival_regions_wrapper.wrapper.abstract_wrapper import AbstractWrapper
-import re
-from bs4 import BeautifulSoup
 
 
-authentication = LocalAuthentication(True)
-authentication.set_credentials("mail@gmail.com", "password", "google")
+class MissingAuthenticationError(Exception):
+    """Error for missing authentication"""
+
+
+RIVAL_REGIONS_USERNAME = os.environ.get("RIVAL_REGIONS_USERNAME", None)
+RIVAL_REGIONS_PASSWORD = os.environ.get("RIVAL_REGIONS_PASSWORD", None)
+RIVAL_REGIONS_LOGIN_METHOD = os.environ.get("RIVAL_REGIONS_LOGIN_METHOD", None)
+CAPTCHA_KEY = os.environ.get("CAPTCHA_KEY", None)
+
+if None in (
+    RIVAL_REGIONS_USERNAME,
+    RIVAL_REGIONS_PASSWORD,
+    RIVAL_REGIONS_LOGIN_METHOD,
+):
+    raise MissingAuthenticationError(
+        "Load the following variables in your user environment: "
+        "RIVAL_REGIONS_USERNAME, RIVAL_REGIONS_PASSWORD, RIVAL_REGIONS_LOGIN_METHOD"
+    )
+
+authentication = LocalAuthentication(False, CAPTCHA_KEY)
+authentication.set_credentials(
+    RIVAL_REGIONS_USERNAME, RIVAL_REGIONS_PASSWORD, RIVAL_REGIONS_LOGIN_METHOD
+)
 
 
 class Market2(AbstractWrapper):