add new var CUSTOM_PIP_PACKAGES

UsergeTeam · Jan 22, 2022 · a23aefb · a23aefb
1 parent d5369ce
commit a23aefb
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/config.env.sample b/config.env.sample
@@ -48,6 +48,10 @@ LOAD_UNOFFICIAL_PLUGINS=false
 ASSERT_SINGLE_INSTANCE=false
 
 
+# Custom pip packages which are seperated by spaces to install while starting the bot.
+CUSTOM_PIP_PACKAGES=""
+
+
 # This is best if you wanna add your own plugins
 # Set this to fork of https://github.com/UsergeTeam/Custom-Plugins and add your plugins
 CUSTOM_PLUGINS_REPO=""

diff --git a/init/checks.sh b/init/checks.sh
@@ -54,6 +54,7 @@ _checkDefaultVars() {
         [UPSTREAM_REPO]="https://github.com/UsergeTeam/Userge"
         [LOAD_UNOFFICIAL_PLUGINS]=false
         [ASSERT_SINGLE_INSTANCE]=false
+        [CUSTOM_PIP_PACKAGES]=""
         [CUSTOM_PLUGINS_REPO]=""
         [G_DRIVE_IS_TD]=true
         [CMD_TRIGGER]="."
@@ -172,6 +173,15 @@ _checkCustomPlugins() {
     _setupPlugins Custom "https://(ghp_[0-9A-z]{36}@)?github.com/.+/.+" $CUSTOM_PLUGINS_REPO
 }
 
+_checkPipPackages() {
+    editLastMessage "Checking Custom Pip Packages ..."
+    if [[ $CUSTOM_PIP_PACKAGES ]]; then
+        log "\tFound and Installing ..."
+        upgradePip
+        installCustomReq "$CUSTOM_PIP_PACKAGES"
+    fi
+}
+
 _flushMessages() {
     deleteLastMessage
 }
@@ -192,5 +202,6 @@ assertEnvironment() {
     _checkUpstreamRepo
     _checkUnoffPlugins
     _checkCustomPlugins
+    _checkPipPackages
     _flushMessages
 }
diff --git a/init/utils.sh b/init/utils.sh
@@ -99,6 +99,10 @@ installReq() {
     pip3 install -r $1/requirements.txt &> /dev/null
 }
 
+installCustomReq() {
+    pip3 install $(tr ' ' '\n' <<< "$1") &> /dev/null
+}
+
 printLine() {
     echo '->- ->- ->- ->- ->- ->- ->- --- -<- -<- -<- -<- -<- -<- -<-'
 }