updated prod deployment to gunicorn

This commit is contained in:
Adrian Gunnar Lauterer 2024-05-31 18:06:33 +02:00
parent 6e61d54c68
commit c6700bd41d
7 changed files with 157 additions and 50 deletions

View File

@ -10,20 +10,80 @@
pkgs = import nixpkgs { system = "x86_64-linux"; };
lib = pkgs.lib;
types = lib.types;
deps = [
pkgs.python3
pkgs.python3Packages.flask
pkgs.python3Packages.flask-socketio
pkgs.python3Packages.requests
pkgs.python3Packages.gunicorn
pkgs.python3Packages.eventlet
];
in
{
packages.x86_64-linux.default = pkgs.python3Packages.buildPythonApplication rec {
packages.x86_64-linux.default = pkgs.python3Packages.buildPythonPackage rec {
pname = "ozai-webui";
version = "0.1.1";
propagatedBuildInputs = [ pkgs.python3Packages.flask pkgs.python3Packages.flask-socketio pkgs.python3Packages.requests ];
#propagatedBuildInputs = deps;
dependencies = deps;
src = ./src;
doCheck = false;
postInstall = ''
mkdir -p $out/share/ozai-webui
cp -r static $out/share/ozai-webui/static
cp -r templates $out/share/ozai-webui/templates
mv $out/bin/main.py $out/bin/ozai-webui
mkdir -p $out/share/ozai_webui/static
mkdir -p $out/share/ozai_webui/templates
install -Dm444 ${src}/static/* $out/share/ozai_webui/static/
install -Dm444 ${src}/templates/* $out/share/ozai_webui/templates/
cp -r $src/templates $out/lib/python3.11/site-packages/ozai_webui/templates
cp -r $src/templates $out/lib/python3.11/site-packages/templates
cp -r $src/templates $out/lib/python3.11/templates
cp -r $src/static $out/lib/python3.11/site-packages/ozai_webui/static
cp -r $src/static $out/lib/python3.11/site-packages/static
cp -r $src/static $out/lib/python3.11/static
'';
};
packages.x86_64-linux.ozai-webui-run = pkgs.stdenv.mkDerivation rec {
pname = "ozai-webui-run";
version = "0.1.1";
src = ./src;
pythonEnv = pkgs.python3.withPackages (ps: with ps; [
flask
flask-socketio
requests
gunicorn
eventlet
]);
buildInputs = [
pkgs.bash
pythonEnv
];
propagatedBuildInputs = buildInputs;
nativeBuildInputs = [
pkgs.makeWrapper
];
installPhase = ''
mkdir -p $out/bin
cp -r $src/templates $out/bin/templates
cp -r $src/static $out/bin/static
cp $src/run.sh $out/bin/run
chmod +x $out/bin/run
wrapProgram $out/bin/run \
--prefix PATH : ${lib.makeBinPath buildInputs}
'';
};
devShells.x86_64-linux.default = pkgs.mkShell {
buildInputs = deps;
shellHook = ''
OZAI_WEBUI_STATIC_FOLDER=result/share/ozai-webui/static
OZAI_WEBUI_TEMPLATE_FOLDER=result/share/ozai-webui/templates
'';
};
@ -45,11 +105,6 @@
default = "http://localhost:8000/api/";
description = "The URL of the Ozai server";
};
secretKey = lib.mkOption {
type = types.str;
default = "secret_key";
description = "The secret key for the Flask app";
};
};
config = lib.mkIf config.services.ozai-webui.enable {
@ -57,8 +112,18 @@
description = "Ozai WebUI server";
after = [ "network.target" ];
wantedBy = [ "multi-user.target" ];
environment = {
OZAI_URL= "${config.services.ozai-webui.ozaiUrl}";
OZAI_WEBUI_HOST= "${config.services.ozai-webui.host}";
OZAI_WEBUI_PORT= toString(config.services.ozai-webui.port);
OZAI_WEBUI_STATIC_FOLDER= "${self.packages.x86_64-linux.default}/share/ozai-webui/static";
OZAI_WEBUI_TEMPLATE_FOLDER="${self.packages.x86_64-linux.default}/share/ozai-webui/templates";
};
serviceConfig = {
ExecStart = "${self.packages.x86_64-linux.default}/bin/ozai-webui --port ${toString config.services.ozai-webui.port} --host '${config.services.ozai-webui.host}' --ozai_url '${config.services.ozai-webui.ozaiUrl}' --secret_key '${config.services.ozai-webui.secretKey}' --static_folder '${self.packages.x86_64-linux.default}/share/ozai-webui/static' --template_folder '${self.packages.x86_64-linux.default}/share/ozai-webui/templates'";
#ExecStart = "${self.packages.x86_64-linux.ozai-webui-run}/bin/run ${config.services.ozai-webui.host} ${toString(config.services.ozai-webui.port)} ${self.packages.x86_64-linux.default}/bin";
ExecStart = "${self.packages.x86_64-linux.ozai-webui-run}/bin/run ${config.services.ozai-webui.host} ${toString(config.services.ozai-webui.port)} ${self.packages.x86_64-linux.default}/lib/python3.11/site-packages/";
#ExecStart = "${self.packages.x86_64-linux.default}/bin/ozai_webui";
Restart = "always";
};
};

1
result Symbolic link
View File

@ -0,0 +1 @@
/nix/store/22y4lw54f6yp05f6h0876m4rwfk7hi0k-ozai-webui-run-0.1.1

1
result-1 Symbolic link
View File

@ -0,0 +1 @@
/nix/store/f0rv69ig250np27kqx6ag8sbyx5f05fy-python3.11-ozai-webui-0.1.1

View File

@ -2,18 +2,15 @@
from flask import Flask, render_template, request
from flask_socketio import SocketIO, join_room
import requests
import argparse
#import argparse
import os
app = Flask(__name__)
socketio = SocketIO(app)
ozai_url = 'http://localhost:8000/api/'
ozai_webui_host = '0.0.0.0'
ozai_webui_port = 5000
app.config['SECRET_KEY'] = "secret"
static_folder = 'static'
template_folder = 'templates'
#static_folder = '../static'
#template_folder = '../templates'
#check if the environment variables are set
if os.getenv('OZAI_URL') is not None:
@ -23,37 +20,48 @@ if os.getenv('OZAI_WEBUI_HOST') is not None:
if os.getenv('OZAI_WEBUI_PORT') is not None:
ozai_webui_port = int(os.getenv('OZAI_WEBUI_PORT'))
if os.getenv('OZAI_WEBUI_SECRET_KEY') is not None:
app.config['SECRET_KEY'] = os.getenv('OZAI_WEBUI_SECRET_KEY')
if os.getenv('OZAI_WEBUI_STATIC_FOLDER') is not None:
static_folder = os.getenv('STATIC_FOLDER')
if os.getenv('OZAI_WEBUI_TEMPLATE_FOLDER') is not None:
template_folder = os.getenv('TEMPLATE_FOLDER')
parser = argparse.ArgumentParser(description="Run the Ozai WebUI server")
parser.add_argument('-H', '--host', type=str, default=ozai_webui_host, help='The host to run the server on')
parser.add_argument('-P', '--port', type=int, default=ozai_webui_port, help='The port to run the server on')
parser.add_argument('-O', '--ozai_url', type=str, default=ozai_url, help='The URL of the Ozai server')
parser.add_argument('-S', '--secret_key', type=str, default=app.config['SECRET_KEY'], help='The secret key for the Flask app')
parser.add_argument('--static_folder', type=str, default=static_folder, help='The location of the static folder')
parser.add_argument('--template_folder', type=str, default=template_folder, help='The location of the template folder')
app.config['OZAI_WEBUI_SECRET_KEY'] = os.getenv('OZAI_WEBUI_SECRET_KEY')
#if os.getenv('OZAI_WEBUI_STATIC_FOLDER') is not None:
# static_folder = os.getenv('OZAI_WEBUI_STATIC_FOLDER')
#if os.getenv('OZAI_WEBUI_TEMPLATE_FOLDER') is not None:
# template_folder = os.getenv('OZAI_WEBUI_TEMPLATE_FOLDER')
args = parser.parse_args()
print(f"using host { ozai_webui_host }")
print(f"using port { ozai_webui_port }")
print(f"using ozai url { ozai_url }")
#print(f"using template folder { template_folder }")
#print(f"using static folder { static_folder }")
if args.host:
ozai_webui_host = args.host
if args.port:
ozai_webui_port = args.port
if args.ozai_url:
ozai_url = args.ozai_url
if args.secret_key:
app.config['SECRET_KEY'] = args.secret_key
if args.static_folder:
app.static_folder = args.static_folder
if args.template_folder:
app.template_folder = args.template_folder
app = Flask(__name__,template_folder="../templates",static_folder="../static" )
socketio = SocketIO(app)
app.config['SECRET_KEY'] = "secret"
#parser = argparse.ArgumentParser(description="Run the Ozai WebUI server")
#
#parser.add_argument('-H', '--host', type=str, default=ozai_webui_host, help='The host to run the server on')
#parser.add_argument('-P', '--port', type=int, default=ozai_webui_port, help='The port to run the server on')
#parser.add_argument('-O', '--ozai_url', type=str, default=ozai_url, help='The URL of the Ozai server')
#parser.add_argument('-S', '--secret_key', type=str, default=app.config['SECRET_KEY'], help='The secret key for the Flask app')
#parser.add_argument('--static_folder', type=str, default=static_folder, help='The location of the static folder')
#parser.add_argument('--template_folder', type=str, default=template_folder, help='The location of the template folder')
#
#
#args = parser.parse_args()
#
#if args.host:
# ozai_webui_host = args.host
#if args.port:
# ozai_webui_port = args.port
#if args.ozai_url:
# ozai_url = args.ozai_url
#if args.secret_key:
# app.config['SECRET_KEY'] = args.secret_key
#if args.static_folder:
# app.static_folder = args.static_folder
#if args.template_folder:
# app.template_folder = args.template_folder
#
#home page
@app.route('/')
@ -210,7 +218,9 @@ def ws_message(data):
#send message to the player that the move was invalid
socketio.emit('move_status', response.text, room=game_id)
def main ():
# app.run(debug=True, host='0.0.0.0', port=5000,ssl_context='adhoc')
return socketio.run(app, debug=False, host=ozai_webui_host, port=ozai_webui_port, allow_unsafe_werkzeug=True)
if __name__ == '__main__':
# app.run(debug=True, host='0.0.0.0', port=5000,ssl_context='adhoc')
socketio.run(app, debug=False, host=ozai_webui_host, port=ozai_webui_port, allow_unsafe_werkzeug=True)
main()

1
src/result Symbolic link
View File

@ -0,0 +1 @@
/nix/store/ldv1rcfs9phhh7hh2wzcvn2600dvgw4y-python3.11-ozai-webui-0.1.1

20
src/run.sh Normal file
View File

@ -0,0 +1,20 @@
#!/usr/bin/env bash
# Default values for host, port, and package path
DEFAULT_HOST="0.0.0.0"
DEFAULT_PORT="8000"
DEFAULT_PACKAGE_PATH="ozai_webui"
DEFAULT_STATIC_PATH="template"
DEFAULT_STATIC_PATH="static"
# Get the host, port, and package path from command line arguments, or use defaults
HOST="${1:-$DEFAULT_HOST}"
PORT="${2:-$DEFAULT_PORT}"
PACKAGE_PATH="${3:-$DEFAULT_PACKAGE_PATH}"
TEMPLATE_PATH="${4:-$DEFAULT_TEMPLATE_PATH}"
STATIC_PATH="${5:-$DEFAULT_STATIC_PATH}"
# Run Gunicorn with the specified host, port, and package path
#gunicorn -w 1 --chdir "$PACKAGE_PATH" --bind "$HOST:$PORT" __init__:main
gunicorn --worker-class eventlet -w 1 --chdir "$PACKAGE_PATH" --bind "$HOST:$PORT" ozai_webui:app

View File

@ -6,6 +6,8 @@ requires = (
"flask",
"flask-socketio",
"requests",
"gunicorn",
"eventlet",
)
@ -17,9 +19,16 @@ setup(
name='ozai-webui',
version='0.1.1',
author = "Adrian Gunnar Lauterer",
packages=find_packages(),
scripts=["main.py"],
packages=find_packages("."),
#scripts=["ozai_webui.py"],
#scripts={"ozai_webui:main": "ozai_webui"},
entry_points = {
"console_scripts": [
"ozai_webui = ozai_webui:main"
]
},
install_requires=requires,
#package_dir= {'':'src'},
package_data={
'': ['templates/*.html', 'static/*.*'],
},