Подсчитайте ожидаемые аргументы, которые будут переданы функции в Python

Проблема

Мне «нужно» подсчитать количество аргументов, которые, как ожидается, будут переданы объекту функции. Мои команды хранятся через их объект, поэтому я могу ссылаться на каждую из них по отдельности. Я уже пробовал инспектировать, и хотя он кажется идеальным, он по какой-то причине не может различить между 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() странным обращением с ним, но я знаю, что обычный пустой список, такой как [], считается за единицу.

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


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

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


person HavenSelph    schedule 07.03.2020    source источник


Ответы (1)


Вам не нужен getfullargspec, если вы обращаетесь только к нулевому элементу.

my_funct.__code__.co_argcount

person user1411571    schedule 07.03.2020