Colton Sellers
c2f0fdc47a
* fix IFS issue in bash docker scripts * Add default image to research config
121 lines
4.2 KiB
Bash
121 lines
4.2 KiB
Bash
#!/bin/bash
|
|
|
|
# QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
|
|
# Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
full_path=$(realpath $0)
|
|
current_dir=$(dirname $full_path)
|
|
default_image=quantconnect/lean:latest
|
|
default_data_dir=$current_dir/Data
|
|
default_results_dir=$current_dir
|
|
default_config_file=$current_dir/Launcher/config.json
|
|
default_python_dir=$current_dir/Algorithm.Python/
|
|
csharp_dll=$current_dir/Launcher/bin/Debug/QuantConnect.Algorithm.CSharp.dll
|
|
csharp_pdb=$current_dir/Launcher/bin/Debug/QuantConnect.Algorithm.CSharp.pdb
|
|
|
|
#If arg is a file process the key values
|
|
if [ -f "$1" ]; then
|
|
IFS="="
|
|
while read -r key value; do
|
|
eval "$key='$value'"
|
|
done < $1
|
|
#If there are in line args, process them
|
|
elif [ ! -z "$*" ]; then
|
|
for arg in "$@"; do
|
|
eval "$arg"
|
|
done
|
|
#Else query user for settings
|
|
else
|
|
read -p "Enter docker image [default: $default_image]: " image
|
|
read -p "Enter absolute path to Lean config file [default: $default_config_file]: " config_file
|
|
read -p "Enter absolute path to Data folder [default: $default_data_dir]: " data_dir
|
|
read -p "Enter absolute path to store results [default: $default_results_dir]: " results_dir
|
|
read -p "Would you like to debug C#? (Requires mono debugger attachment) [default: N]: " debugging
|
|
fi
|
|
|
|
#Have to reset IFS for cfg files to work properly
|
|
IFS=" "
|
|
|
|
if [ -z "$image" ]; then
|
|
image=$default_image
|
|
fi
|
|
|
|
if [ -z "$config_file" ]; then
|
|
config_file=$default_config_file
|
|
fi
|
|
|
|
if [ -z "$python_dir" ]; then
|
|
python_dir=$default_python_dir
|
|
fi
|
|
|
|
if [ ! -f "$config_file" ]; then
|
|
echo "Lean config file $config_file does not exist"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z "$data_dir" ]; then
|
|
data_dir=$default_data_dir
|
|
fi
|
|
|
|
if [ ! -d "$data_dir" ]; then
|
|
echo "Data directory $data_dir does not exist"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z "$results_dir" ]; then
|
|
results_dir=$default_results_dir
|
|
fi
|
|
|
|
if [ ! -d "$results_dir" ]; then
|
|
echo "Results directory $results_dir does not exist"
|
|
exit 1
|
|
fi
|
|
|
|
#First part of the docker command that is static, then we build the rest
|
|
command="docker run --rm --mount type=bind,source=$config_file,target=/Lean/Launcher/config.json,readonly \
|
|
--mount type=bind,source=$data_dir,target=/Data,readonly \
|
|
--mount type=bind,source=$results_dir,target=/Results \
|
|
--name LeanEngine \
|
|
-p 5678:5678 "
|
|
|
|
#If the csharp dll and pdb are present, mount them
|
|
if [ ! -f "$csharp_dll" ]; then
|
|
echo "Csharp file at '$csharp_dll' does not exist; no CSharp files will be mounted"
|
|
else
|
|
command+="--mount type=bind,source=$csharp_dll,target=/Lean/Launcher/bin/Debug/QuantConnect.Algorithm.CSharp.dll \
|
|
--mount type=bind,source=$csharp_pdb,target=/Lean/Launcher/bin/Debug/QuantConnect.Algorithm.CSharp.pdb "
|
|
fi
|
|
|
|
#If python algorithms are present, mount them
|
|
if [ ! -d "$python_dir" ]; then
|
|
echo "No Python Algorithm location found at '$python_dir'; no Python files will be mounted"
|
|
else
|
|
command+="--mount type=bind,source=$python_dir,target=/Lean/Algorithm.Python "
|
|
fi
|
|
|
|
#If debugging is set then set the entrypoint to run mono with a debugger server
|
|
shopt -s nocasematch
|
|
if [[ "$debugging" == "y" ]]; then
|
|
command+="-p 55555:55555 \
|
|
--entrypoint mono \
|
|
$image --debug --debugger-agent=transport=dt_socket,server=y,address=0.0.0.0:55555,suspend=y \
|
|
QuantConnect.Lean.Launcher.exe --data-folder /Data --results-destination-folder /Results --config /Lean/Launcher/config.json"
|
|
|
|
echo "Docker container starting, attach to Mono process at localhost:55555 to begin"
|
|
else
|
|
command+="$image --data-folder /Data --results-destination-folder /Results --config /Lean/Launcher/config.json"
|
|
fi
|
|
|
|
#Run built docker command; docker requires sudo privledges by default
|
|
eval sudo $command
|