ED_LRR/rust/multi_test.py

47 lines
1.3 KiB
Python

import os
def setup_logging(loglevel="INFO"):
import logging
import coloredlogs
import datetime
coloredlogs.DEFAULT_FIELD_STYLES["delta"] = {"color": "green"}
coloredlogs.DEFAULT_FIELD_STYLES["levelname"] = {"color": "yellow"}
class DeltaTimeFormatter(coloredlogs.ColoredFormatter):
def format(self, record):
seconds = record.relativeCreated / 1000
duration = datetime.timedelta(seconds=seconds)
record.delta = str(duration)
return super().format(record)
coloredlogs.ColoredFormatter = DeltaTimeFormatter
logfmt = " | ".join(
["[%(delta)s] %(levelname)s", "%(name)s:%(pathname)s:%(lineno)s", "%(message)s"]
)
numeric_level = getattr(logging, loglevel.upper(), None)
if not isinstance(numeric_level, int):
raise ValueError("Invalid log level: %s" % loglevel)
coloredlogs.install(level=numeric_level, fmt=logfmt)
setup_logging()
_ed_lrr = __import__("_ed_lrr")
r = _ed_lrr.PyRouter(None)
r.load("stars.csv")
# r.run_bfs(48)
r.test(48)
exit()
_ed_lrr.PyRouter.preprocess_galaxy("E:/EDSM/galaxy.json.gz", "E:/EDSM/stars.csv")
exit()
r = _ed_lrr.PyRouter(print)
r.load("../stars.csv")
systems = r.resolve_systems((0, 0, 0), "Colonia", 18627)
print(systems)
print(systems[0, 0, 0])