Mirrored_Repos/crafty-4
1
0
Fork 0
mirror of https://gitlab.com/crafty-controller/crafty-4.git synced 2024-08-30 18:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add messages for loading screen

Browse Source
This commit is contained in:
amcmanu3 2023-11-03 16:41:37 -04:00
parent 28b3a4f8f3
commit 61c1621cbb
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
main.py
View File

@ -109,11 +109,12 @@ if __name__ == "__main__":
) )
args = parser.parse_args() args = parser.parse_args()
helper.ensure_logging_setup() helper.ensure_logging_setup()
helper.crafty_starting = True
# Init WebSocket Manager Here
ws = WebSocketManager()
ws.broadcast("update", {"message": "Setting Up Logging"})
setup_logging(debug=args.verbose) setup_logging(debug=args.verbose)
if args.verbose: if args.verbose:
Console.level = "debug" Console.level = "debug"
@ -126,15 +127,18 @@ if __name__ == "__main__":
# print our pretty start message # print our pretty start message
do_intro() do_intro()
ws.broadcast("update", {"message": "Securing Session For Process"})
# our session file, helps prevent multiple controller agents on the same machine. # our session file, helps prevent multiple controller agents on the same machine.
helper.create_session_file(ignore=args.ignore) helper.create_session_file(ignore=args.ignore)
ws.broadcast("update", {"message": "Initializing Database"})
# start the database # start the database
database = peewee.SqliteDatabase( database = peewee.SqliteDatabase(
helper.db_path, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10} helper.db_path, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10}
) )
database_proxy.initialize(database) database_proxy.initialize(database)
ws.broadcast("update", {"message": "Checking For Database Migrations"})
migration_manager = MigrationManager(database, helper) migration_manager = MigrationManager(database, helper)
migration_manager.up() # Automatically runs migrations migration_manager.up() # Automatically runs migrations
@ -165,8 +169,6 @@ if __name__ == "__main__":
Console.info("No flag found. Secrets are staying") Console.info("No flag found. Secrets are staying")
file_helper = FileHelpers(helper) file_helper = FileHelpers(helper)
import_helper = ImportHelpers(helper, file_helper) import_helper = ImportHelpers(helper, file_helper)
# Init WebSocket Manager Here
WebSocketManager()
# now the tables are created, we can load the tasks_manager and server controller # now the tables are created, we can load the tasks_manager and server controller
controller = Controller(database, helper, file_helper, import_helper) controller = Controller(database, helper, file_helper, import_helper)
Console.info("Checking for remote changes to config.json") Console.info("Checking for remote changes to config.json")
@ -174,6 +176,7 @@ if __name__ == "__main__":
Console.info("Remote change complete.") Console.info("Remote change complete.")
import3 = Import3(helper, controller) import3 = Import3(helper, controller)
ws.broadcast("update", {"message": "Starting Task Scheduler"})
tasks_manager = TasksManager(helper, controller, file_helper) tasks_manager = TasksManager(helper, controller, file_helper)
tasks_manager.start_webserver() tasks_manager.start_webserver()
@ -225,6 +228,7 @@ if __name__ == "__main__":
"the server may be limited." "the server may be limited."
) )
ws.broadcast("update", {"message": "Checking For Internet"})
internet_check_thread = Thread(target=internet_check, name="internet_check") internet_check_thread = Thread(target=internet_check, name="internet_check")
def controller_setup(): def controller_setup():
@ -272,7 +276,7 @@ if __name__ == "__main__":
time.sleep(0.01) # Wait for the daemon info message time.sleep(0.01) # Wait for the daemon info message
Console.info("Setting up Crafty's internal components...") Console.info("Setting up Crafty's internal components...")
ws.broadcast("update", {"message": "Final checks and launching"})
# Start the setup threads # Start the setup threads
tasks_starter_thread.start() tasks_starter_thread.start()
internet_check_thread.start() internet_check_thread.start()
@ -282,8 +286,9 @@ if __name__ == "__main__":
tasks_starter_thread.join() tasks_starter_thread.join()
internet_check_thread.join() internet_check_thread.join()
controller_setup_thread.join() controller_setup_thread.join()
ws.broadcast("send_start_reload", "")
Console.info("Crafty has fully started and is now ready for use!") Console.info("Crafty has fully started and is now ready for use!")
helper.crafty_starting = False
# Check if new version available # Check if new version available
remote_ver = helper.check_remote_version() remote_ver = helper.check_remote_version()