mirror of
https://github.com/RYDE-WORK/ballistica.git
synced 2026-01-22 23:15:49 +08:00
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
# Released under the MIT License. See LICENSE for details.
|
|
#
|
|
"""Message related tools functionality."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING
|
|
|
|
from efrotools.code import format_yapf_str
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Set, List, Dict, Any, Union, Optional
|
|
|
|
|
|
def standard_message_sender_gen_pcommand(
|
|
projroot: Path,
|
|
basename: str,
|
|
source_module: str,
|
|
enable_sync_sends: bool,
|
|
enable_async_sends: bool,
|
|
) -> None:
|
|
"""Used by pcommands taking a single filename argument."""
|
|
|
|
import efro.message
|
|
from efro.terminal import Clr
|
|
from efro.error import CleanError
|
|
|
|
if len(sys.argv) != 3:
|
|
raise CleanError('Expected 1 arg: out-path.')
|
|
|
|
dst = sys.argv[2]
|
|
out = format_yapf_str(
|
|
projroot,
|
|
efro.message.create_sender_module(
|
|
basename,
|
|
protocol_create_code=(f'from {source_module} import get_protocol\n'
|
|
f'protocol = get_protocol()'),
|
|
enable_sync_sends=enable_sync_sends,
|
|
enable_async_sends=enable_async_sends,
|
|
))
|
|
|
|
print(f'Meta-building {Clr.BLD}{dst}{Clr.RST}')
|
|
Path(dst).parent.mkdir(parents=True, exist_ok=True)
|
|
with open(dst, 'w', encoding='utf-8') as outfile:
|
|
outfile.write(out)
|
|
|
|
|
|
def standard_message_receiver_gen_pcommand(
|
|
projroot: Path,
|
|
basename: str,
|
|
source_module: str,
|
|
is_async: bool,
|
|
) -> None:
|
|
"""Used by pcommands generating efro.message receiver modules."""
|
|
|
|
import efro.message
|
|
from efro.terminal import Clr
|
|
from efro.error import CleanError
|
|
|
|
if len(sys.argv) != 3:
|
|
raise CleanError('Expected 1 arg: out-path.')
|
|
|
|
dst = sys.argv[2]
|
|
out = format_yapf_str(
|
|
projroot,
|
|
efro.message.create_receiver_module(
|
|
basename,
|
|
protocol_create_code=(f'from {source_module} import get_protocol\n'
|
|
f'protocol = get_protocol()'),
|
|
is_async=is_async,
|
|
))
|
|
|
|
print(f'Meta-building {Clr.BLD}{dst}{Clr.RST}')
|
|
Path(dst).parent.mkdir(parents=True, exist_ok=True)
|
|
with open(dst, 'w', encoding='utf-8') as outfile:
|
|
outfile.write(out)
|