Detecting gaps in requirements.txt and offering to install.

Author: tarehart

rlbot_gui/gui.py

@@ -1,5 +1,6 @@
 import os
 import shutil
+from pathlib import Path
 
 import eel
 from PyQt5.QtCore import QSettings
@@ -8,14 +9,16 @@
 from rlbot.parsing.agent_config_parser import create_looks_configurations, BOT_CONFIG_LOADOUT_HEADER, \
     BOT_CONFIG_LOADOUT_ORANGE_HEADER, BOT_CONFIG_LOADOUT_PAINT_BLUE_HEADER, BOT_CONFIG_LOADOUT_PAINT_ORANGE_HEADER, \
     load_bot_appearance
-from rlbot.parsing.bot_config_bundle import get_bot_config_bundle, get_script_config_bundle, RunnableConfigBundle
+from rlbot.parsing.bot_config_bundle import get_bot_config_bundle, get_script_config_bundle, RunnableConfigBundle, \
+    BotConfigBundle
 from rlbot.parsing.directory_scanner import scan_directory_for_bot_configs, scan_directory_for_script_configs
 from rlbot.parsing.match_settings_config_parser import map_types, game_mode_types, \
     boost_amount_mutator_types, match_length_types, max_score_types, overtime_mutator_types, \
     series_length_mutator_types, game_speed_mutator_types, ball_max_speed_mutator_types, ball_type_mutator_types, \
     ball_weight_mutator_types, ball_size_mutator_types, ball_bounciness_mutator_types, rumble_mutator_types, \
     boost_strength_mutator_types, gravity_mutator_types, demolish_mutator_types, respawn_time_mutator_types, \
     existing_match_behavior_types
+from rlbot.utils.requirements_management import install_requirements_file
 
 from rlbot_gui.bot_management.bot_creation import bootstrap_python_bot, bootstrap_scratch_bot, \
     bootstrap_python_hivemind, convert_to_filename
@@ -68,7 +71,7 @@ def pick_bot_folder():
     return []
 
 
-def serialize_bundle(bundle):
+def serialize_bundle(bundle: BotConfigBundle):
     return {
         'name': bundle.name,
         'type': 'rlbot',
@@ -77,7 +80,8 @@ def serialize_bundle(bundle):
         'path': bundle.config_path,
         'looks_path': bundle.looks_path,
         'info': read_info(bundle),
-        'logo': try_copy_logo(bundle)
+        'logo': try_copy_logo(bundle),
+        'missing_python_packages': [r.name for r in bundle.get_missing_python_packages()],
     }
 
 
@@ -88,7 +92,8 @@ def serialize_script_bundle(bundle):
         'image': 'imgs/rlbot.png',
         'path': bundle.config_path,
         'info': read_info(bundle),
-        'logo': try_copy_logo(bundle)
+        'logo': try_copy_logo(bundle),
+        'missing_python_packages': [r.name for r in bundle.get_missing_python_packages()],
     }
 
 
@@ -374,6 +379,20 @@ def install_package(package_string):
     return {'exitCode': exit_code, 'package': package_string}
 
 
+@eel.expose
+def install_requirements(config_path):
+    try:
+        bundle = get_bot_config_bundle(config_path)
+    except Exception:
+        bundle = get_script_config_bundle(config_path)
+
+    if bundle.requirements_file:
+        exit_code = install_requirements_file(bundle.requirements_file)
+        return {'exitCode': exit_code, 'package': bundle.requirements_file}
+    else:
+        return {'exitCode': 1, 'package': None}
+
+
 def get_last_botpack_commit_id():
     url = f'https://api.github.com/repos/{BOTPACK_REPO_OWNER}/{BOTPACK_REPO_NAME}/branches/{BOTPACK_REPO_BRANCH}'
     try:

rlbot_gui/gui/main.vue

@@ -116,6 +116,10 @@
 			</draggable>
 			<md-card class="bot-card md-elevation-3" v-for="script in scriptPool">
 				<md-switch v-model="script.enabled">{{script.name}}</md-switch>
+				<md-button class="md-icon-button md-dense warning-icon" v-if="script.warn"
+						   @click.stop="activeBot = script; showLanguageWarning = true;">
+					<md-icon>warning</md-icon>
+				</md-button>
 				<md-button class="md-icon-button md-dense bot-hover-reveal" v-if="script.info"
 						   @click.stop="activeBot = script; showBotInfo = true;">
 					<md-icon>blur_on</md-icon>
@@ -338,6 +342,14 @@
 						<a href="https://www.google.com/chrome/" target="_blank">download it here</a>.
 					</p>
 				</div>
+				<div v-if="activeBot.warn === 'pythonpkg'">
+					<p>
+						This bot needs some python packages you haven't installed yet:
+						<code><span v-for="missing in activeBot.missing_python_packages">{{missing}} </span></code>
+					</p>
+					<md-button @click="installRequirements(activeBot.path)"
+							   class="md-primary md-raised">Install Now</md-button>
+				</div>
 			</md-dialog-content>
 
 			<md-dialog-actions>
@@ -675,13 +687,13 @@
 				freshScripts.forEach((script) => {script.enabled = !!this.matchSettings.scripts.find( (element) => element.path === script.path )});
 
 				this.scriptPool = this.scriptPool.concat(freshScripts).sort((a, b) => a.name.localeCompare(b.name));
-
+				this.applyLanguageWarnings();
 				this.showProgressSpinner = false;
 			},
 
 	 		applyLanguageWarnings: function () {
 				if (this.languageSupport) {
-					this.botPool.forEach((bot) => {
+					this.botPool.concat(this.scriptPool).forEach((bot) => {
 						if (bot.info && bot.info.language) {
 							const language = bot.info.language.toLowerCase();
 							if (!this.languageSupport.java && language.match(/java|kotlin|scala/)) {
@@ -691,6 +703,10 @@
 								bot.warn = 'chrome';
 							}
 						}
+						console.log(bot);
+						if (bot.missing_python_packages && bot.missing_python_packages.length > 0) {
+							bot.warn = 'pythonpkg';
+						}
 					});
 				}
 			},
@@ -741,6 +757,10 @@
 				this.showProgressSpinner = true;
 				eel.install_package(this.packageString)(this.onInstallationComplete);
 			},
+			installRequirements: function (configPath) {
+				this.showProgressSpinner = true;
+				eel.install_requirements(configPath)(this.onInstallationComplete);
+			}
 		},
 		created: function () {
 			eel.get_folder_settings()(this.folderSettingsReceived);