init: first version
This commit is contained in:
commit
966d4819b4
141
.gitignore
vendored
Normal file
141
.gitignore
vendored
Normal file
@ -0,0 +1,141 @@
|
|||||||
|
# Byte-compiled / optimized / DLL files
|
||||||
|
__pycache__/
|
||||||
|
*.py[cod]
|
||||||
|
*$py.class
|
||||||
|
|
||||||
|
# C extensions
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Distribution / packaging
|
||||||
|
.Python
|
||||||
|
build/
|
||||||
|
develop-eggs/
|
||||||
|
dist/
|
||||||
|
downloads/
|
||||||
|
eggs/
|
||||||
|
.eggs/
|
||||||
|
lib/
|
||||||
|
lib64/
|
||||||
|
parts/
|
||||||
|
sdist/
|
||||||
|
var/
|
||||||
|
wheels/
|
||||||
|
share/python-wheels/
|
||||||
|
*.egg-info/
|
||||||
|
.installed.cfg
|
||||||
|
*.egg
|
||||||
|
MANIFEST
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
# Usually these files are written by a python script from a template
|
||||||
|
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# Installer logs
|
||||||
|
pip-log.txt
|
||||||
|
pip-delete-this-directory.txt
|
||||||
|
|
||||||
|
# Unit test / coverage reports
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
.cache
|
||||||
|
nosetests.xml
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
*.py,cover
|
||||||
|
.hypothesis/
|
||||||
|
.pytest_cache/
|
||||||
|
cover/
|
||||||
|
|
||||||
|
# Translations
|
||||||
|
*.mo
|
||||||
|
*.pot
|
||||||
|
|
||||||
|
# Django stuff:
|
||||||
|
*.log
|
||||||
|
local_settings.py
|
||||||
|
db.sqlite3
|
||||||
|
db.sqlite3-journal
|
||||||
|
|
||||||
|
# Flask stuff:
|
||||||
|
instance/
|
||||||
|
.webassets-cache
|
||||||
|
|
||||||
|
# Scrapy stuff:
|
||||||
|
.scrapy
|
||||||
|
|
||||||
|
# Sphinx documentation
|
||||||
|
docs/_build/
|
||||||
|
|
||||||
|
# PyBuilder
|
||||||
|
.pybuilder/
|
||||||
|
target/
|
||||||
|
|
||||||
|
# Jupyter Notebook
|
||||||
|
.ipynb_checkpoints
|
||||||
|
|
||||||
|
# IPython
|
||||||
|
profile_default/
|
||||||
|
ipython_config.py
|
||||||
|
|
||||||
|
# pyenv
|
||||||
|
# For a library or package, you might want to ignore these files since the code is
|
||||||
|
# intended to run in multiple environments; otherwise, check them in:
|
||||||
|
# .python-version
|
||||||
|
|
||||||
|
# pipenv
|
||||||
|
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||||
|
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||||
|
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||||
|
# install all needed dependencies.
|
||||||
|
#Pipfile.lock
|
||||||
|
|
||||||
|
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
|
||||||
|
__pypackages__/
|
||||||
|
|
||||||
|
# Celery stuff
|
||||||
|
celerybeat-schedule
|
||||||
|
celerybeat.pid
|
||||||
|
|
||||||
|
# SageMath parsed files
|
||||||
|
*.sage.py
|
||||||
|
|
||||||
|
# Environments
|
||||||
|
.env
|
||||||
|
.venv
|
||||||
|
env/
|
||||||
|
venv/
|
||||||
|
ENV/
|
||||||
|
env.bak/
|
||||||
|
venv.bak/
|
||||||
|
|
||||||
|
# Spyder project settings
|
||||||
|
.spyderproject
|
||||||
|
.spyproject
|
||||||
|
|
||||||
|
# Rope project settings
|
||||||
|
.ropeproject
|
||||||
|
|
||||||
|
# mkdocs documentation
|
||||||
|
/site
|
||||||
|
|
||||||
|
# mypy
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
dmypy.json
|
||||||
|
|
||||||
|
# Pyre type checker
|
||||||
|
.pyre/
|
||||||
|
|
||||||
|
# pytype static type analyzer
|
||||||
|
.pytype/
|
||||||
|
|
||||||
|
# Cython debug symbols
|
||||||
|
cython_debug/
|
||||||
|
|
||||||
|
# Pycharm
|
||||||
|
.idea
|
69
debug.py
Executable file
69
debug.py
Executable file
@ -0,0 +1,69 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# -*- mode: Python -*-
|
||||||
|
|
||||||
|
import os
|
||||||
|
import socket
|
||||||
|
import sys
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
'''
|
||||||
|
AI director debugger
|
||||||
|
|
||||||
|
Interactive and dynamic debug user interface
|
||||||
|
|
||||||
|
* The debugger server opens a local socket connection for realtime commands
|
||||||
|
* The debugger server has a many methods available (list, set, etc.)
|
||||||
|
* The user can activate tags: the debugger will output messages with active flag at reception
|
||||||
|
* The user can debug objects: the object will output part or all of its internals
|
||||||
|
|
||||||
|
Licensed under GNU GPLv3
|
||||||
|
|
||||||
|
by alban
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
valid_commands = {
|
||||||
|
'list': {
|
||||||
|
"api": "list_tags",
|
||||||
|
"help": "List available tags",
|
||||||
|
"tags": None
|
||||||
|
},
|
||||||
|
'add': {
|
||||||
|
"api": "add_tags",
|
||||||
|
"help": "Debug the given tag(s)",
|
||||||
|
"tags": "tags"
|
||||||
|
},
|
||||||
|
'del': {
|
||||||
|
"api": "def_tags",
|
||||||
|
"help": "Remove from debug the given tag(s)",
|
||||||
|
"tags": "tags"
|
||||||
|
},
|
||||||
|
'help': {
|
||||||
|
"api": "help",
|
||||||
|
"help": "Request help",
|
||||||
|
"tags": None
|
||||||
|
},
|
||||||
|
}
|
||||||
|
socket_addr = "/tmp/._my_debug_socket"
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser(description="AI Director debugger utility")
|
||||||
|
subparsers = parser.add_subparsers(dest='command', help="API request")
|
||||||
|
for i in valid_commands:
|
||||||
|
sub = subparsers.add_parser(i)
|
||||||
|
item = valid_commands[i]
|
||||||
|
if item["tags"]:
|
||||||
|
sub.add_argument("tags", type=str, nargs="+", help=item["help"])
|
||||||
|
parser.add_argument("--socket", help="Socket location. Default: '{}' ".format(socket_addr), default=socket_addr,
|
||||||
|
required=False)
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.command not in valid_commands:
|
||||||
|
raise Exception("Invalid command '{}' not in {}".format(args.command, ", ".join(valid_commands)))
|
||||||
|
command = valid_commands[args.command]["api"]
|
||||||
|
tags = " ".join(args.tags) if "tags" in args else ""
|
||||||
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||||
|
s.connect(socket_addr)
|
||||||
|
s.send(bytes("{} {}".format(command,tags), "utf-8"))
|
||||||
|
data = s.recv(1024)
|
||||||
|
print('Received', str(data, "utf-8"))
|
55
lib/planning.py
Normal file
55
lib/planning.py
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
from debuggable import Debuggable
|
||||||
|
from schedule import Schedule
|
||||||
|
|
||||||
|
|
||||||
|
def check_schedules(schedules):
|
||||||
|
for schedule in schedules:
|
||||||
|
if isinstance(schedule, Schedule):
|
||||||
|
continue
|
||||||
|
raise Exception("Invalid argument provided: {}".format(schedule))
|
||||||
|
|
||||||
|
|
||||||
|
class Planning(Debuggable):
|
||||||
|
|
||||||
|
schedules: list = []
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super(Planning, self).__init__()
|
||||||
|
|
||||||
|
def add(self, schedules):
|
||||||
|
try:
|
||||||
|
# check if all are valid schedules
|
||||||
|
check_schedules(schedules)
|
||||||
|
# clean the schedules for elapsed
|
||||||
|
self.clean_schedules()
|
||||||
|
# add the schedules
|
||||||
|
self.debug("planning_add_schedules", "Adding {} schedules".format(len(schedules)))
|
||||||
|
self.schedules.extend(schedules)
|
||||||
|
self.debug("planning_add_schedules", "Got {} schedules after".format(len(self.schedules)))
|
||||||
|
# sort
|
||||||
|
self.schedules.sort(key=lambda x: x.start_at)
|
||||||
|
|
||||||
|
except Exception as exc:
|
||||||
|
print(exc)
|
||||||
|
|
||||||
|
def clean_schedules(self):
|
||||||
|
removed = 0
|
||||||
|
self.debug("planning_clean_schedules","Got {} schedules before cleaning".format(len(self.schedules)))
|
||||||
|
schedule_length = len(self.schedules)
|
||||||
|
for j in range(0,schedule_length):
|
||||||
|
i = schedule_length - j - 1
|
||||||
|
schedule = self.schedules[i]
|
||||||
|
if schedule.elapsed():
|
||||||
|
del(self.schedules[i])
|
||||||
|
self.debug("planning_clean_schedules_indexes", "Removed index {}/{}".format(i,schedule_length))
|
||||||
|
removed += 1
|
||||||
|
self.debug("planning_clean_schedules","Removed {} schedules".format(removed))
|
||||||
|
|
||||||
|
def get_last_schedule(self):
|
||||||
|
end_at_ordered = self.schedules.copy()
|
||||||
|
if not end_at_ordered:
|
||||||
|
return None
|
||||||
|
end_at_ordered.sort(key=lambda x: x.end_at)
|
||||||
|
return end_at_ordered.pop() if len(self.schedules) else None
|
||||||
|
|
||||||
|
# EOF
|
59
main.py
Executable file
59
main.py
Executable file
@ -0,0 +1,59 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# -*- mode: Python -*-
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
v0.0.0
|
||||||
|
|
||||||
|
AI director
|
||||||
|
|
||||||
|
Licensed under GNU GPLv3
|
||||||
|
|
||||||
|
by alban
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
from os import path, getcwd
|
||||||
|
import sys
|
||||||
|
abspath, filename = path.split(path.realpath(__file__ ))
|
||||||
|
sys.path.insert(0, path.join(abspath,"lib"))
|
||||||
|
import argparse
|
||||||
|
import asyncio
|
||||||
|
import atexit
|
||||||
|
import debugger
|
||||||
|
from lib.director import Director
|
||||||
|
from lib.my_plugin import MyPlugin
|
||||||
|
# from actor_redis import RedisActor
|
||||||
|
# from sensor_redis import RedisSensor
|
||||||
|
# from planner_dummy import DummyPlanner
|
||||||
|
# from planner_duration import DurationPlanner
|
||||||
|
# from planner_scheduler import SchedulerPlanner
|
||||||
|
from lib.planner_content import ContentPlanner
|
||||||
|
|
||||||
|
debugger.init()
|
||||||
|
|
||||||
|
def exit_handler():
|
||||||
|
print('\n\nLaunch resources cleanup before exit.\n\n')
|
||||||
|
debugger.instance.cleanup()
|
||||||
|
|
||||||
|
|
||||||
|
atexit.register(exit_handler)
|
||||||
|
|
||||||
|
argsparser = argparse.ArgumentParser(description="AI Director main script")
|
||||||
|
argsparser.add_argument("--tags","-t", nargs='+', help="Default Debug tags", default="default")
|
||||||
|
args = argsparser.parse_args()
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
director = Director(debug_tags=args.tags)
|
||||||
|
director.add_plugin(MyPlugin())
|
||||||
|
# director.add_plugin(audioPlugin())
|
||||||
|
server = await debugger.instance.server()
|
||||||
|
async with server:
|
||||||
|
await asyncio.gather(
|
||||||
|
server.serve_forever(),
|
||||||
|
director.run()
|
||||||
|
)
|
||||||
|
|
||||||
|
asyncio.run(main())
|
Loading…
Reference in New Issue
Block a user