Подсчитать ожидаемые аргументы, которые будут переданы функции в Python - PullRequest
2 голосов
/ 08 марта 2020

Проблема

Мне нужно подсчитать количество аргументов, которые, как ожидается, будут переданы функциональному объекту. Мои команды хранятся через их объект, поэтому я могу ссылаться на каждую в отдельности. Я уже пробовал проверять и, хотя он кажется идеальным, он по какой-то причине не может различить guish между 0 аргументами и 1 аргументом как оба вывода 1. Кажется, мне придется связать реальный код как это должно быть ошибка с моим личным сценарием. Вот код:

# ============== Impots ==============
from shlex import split
from inspect import getfullargspec

# ==============  Main  ==============
class Command:
  def __init__(self, fn, name=None, aliases=None):
    self.name = name or fn.__name__
    self.aliases = aliases or []
    self.fn = fn

  def __call__(self, *args, **kwargs):
    return self.fn(*args, **kwargs)

  @property
  def all_names(self):
    return (self.name, *self.aliases)


# ============= Storage =============
class Commands(dict):
  def __init__(self, not_found: str='{} was not recognized.', argmismatch: str='{} takes {} arguments but {} were given.'):
    self.flag = [not_found, argmismatch]
    super().__init__()

  def _add_command(self, command):
    for cmd_name in command.all_names:
      if cmd_name in self:
        raise ValueError(f'Name or alias assigned to function {command.fn.__name__} is duplicate: {cmd_name}')
      self[cmd_name] = command

  def add_command(self, name=None, aliases=None):
    def inner_fn(fn):
      self._add_command(Command(fn, name, aliases))
      return fn
    return inner_fn

  def execute(self, user_input):
    self.command, *args = self.parse(user_input or 'no_input')
    if self.command.lower() in self:
      # THIS IS WHERE THE ISSUE HAPPENS
      print(f'Passed: {len(args)}')
      print(f'Expected: {len(getfullargspec(self[self.command])[0])}')
      if not (len(args)==len(getfullargspec(self[self.command])[0])):
        return (False, self.flag[1].format(self.command, len(getfullargspec(self[self.command])[0]), len(args)))
      self[self.command](*args)
      return (True, 'Command found!')
    else:
      return (False, self.flag[0].format(self.command))

  @staticmethod
  def parse(string):
    if (string=='no_input'):
      return ['','']
    return split(string)




cmd = Commands()
@cmd.add_command(name='foo')
def foo():
  print('bar')
cmd.execute('foo bar')

Не проблема

  • Я уже пытался увидеть, был ли список заполнен значением Null, это не так.
  • Он делает это на разных платформах (VSCode, Repl.it, PyDroid), так что это не та странная проблема.
  • Я не уверен, что это может быть len() странным образом, но я знаю, что обычный пустой список, такой как [], считается как один.

Я использую это для сообщения об ошибке и, возможно, для автоматических подсказок о том, что могло быть не так с Команда, я хотел бы любую помощь и буду обновлять мой пост с новой информацией и возможных, правильных и неправильных решений.


Похожие сообщения

Это сообщение является похоже на вопрос, но предоставленное решение не соответствует тому, что мне требуется.

1 Ответ

1 голос
/ 08 марта 2020

Вам не нужно getfullargspec, если вы получаете доступ только к нулевому элементу.

my_funct.__code__.co_argcount

...