Python: クラスのgetメソッドの戻り値の型について

Question

前提・実現したいこと

コンピュータシステムの理論と実装を学習中です。現在8章でVM変換器をPythonで実装しようとしています。以下コードの一部です。

## f_parser.py
import re

class Parser():
    def __init__(self, i_stream):
        self.stream = i_stream
        self.line_now = '\n'
        self.commands = None
        self.match_obj = None
        self.command_type = None
        self.comment = None
        self.command_reg = re.compile('([a-zA-Z]+)?\s?([a-zA-Z0-9]+)?\s?([a-zA-Z0-9]+)?\s?(^(//){1}.*)?\s?')

        self.command_dict = {'add': 'C_ARITHMETRIC',
                            'sub': 'C_ARITHMETRIC',
                            'neg': 'C_ARITHMETRIC',
                            'eq': 'C_ARITHMETRIC',
                            'gt': 'C_ARITHMETRIC',
                            'lt': 'C_ARITHMETRIC',
                            'and': 'C_ARITHMETRIC',
                            'or': 'C_ARITHMETRIC',
                            'not': 'C_ARITHMETRIC',
                            'push': 'C_PUSH',
                            'pop': 'C_POP',
                            'label': 'C_LABEL',
                            'goto': 'C_GOTO',
                            'if-goto' : 'C_IF',
                            'function' : 'C_FUNCTION',
                            'return': 'C_RETURN',
                            'call': 'C_CALL'}


    def __del__(self):
        self.stream.close()


    def hasMoreCommands(self):
        return self.line_now != ''
    
    def advance(self):
        self.line_now = self.stream.readline()
    
    def commandType(self):
        self.match_obj = self.command_reg.match(self.line_now)
        self.commands = [self.match_obj.group(1),
                        self.match_obj.group(2),
                        self.match_obj.group(3),
                        self.match_obj.group(4)]

        self.command_type = self.command_dict.get(self.commands[0])
        return self.command_type


    def get_arg1(self):
        if self.command_type == 'C_ARITHMETRIC':
            return self.commands[0]
        else:
            return self.commands[1]
        

    def get_arg2(self):
        if self.command_type in ['C_PUSH', 'C_POP', 'C_FUNCTION', 'C_CALL']:
            return self.commands[2]
        else:
            return None
        

実現したいことは、push constant 2 のようなVMコードをパースして以下のような関数に投げて処理がしたいです。

## code_writer.py

def writePush(self, arg1, arg2):
    if arg1 == 'static':
        address = ['@' + self.filename + '.' + arg2]
        process = ['D=M']

    elif arg1 == 'constant':
        address = ['@' + str(self.mapping.get(arg1) + int(arg2))]
        process = ['D=A']
        
    elif arg1 in ['pointer', 'temp']:
            address = ['@' + str(self.mapping.get(arg1))]
            process = ['A=M' ,'A=A+{0}'.format(arg2), 'D=M']
        
    elif arg1 in ['local', 'argument', 'this', 'that']:
        address = ['@' + self.mapping.get(arg1)]
        process = ['A=M', 'A=A+{0}'.format(arg2), 'D=M']

            
    command = '\n'.join(address + process)
    self.o_stream.write(command)

get_arg1,get_arg2はf_parser.pyに記述されています。やりたい処理は同じディレクトリ内のcode_writer.pyに書いてあります。実際にはWritePush()は、vm_translator.pyというファイル(これも同じディレクトリ内)でVM_Translatorクラスのメソッド内の処理として呼び出されています。

# vm_translator.py

import f_parser
import code_writer


class VM_Translator():
    def __init__(self, dir, filename):
        self.dir = dir
        self.filename = filename
        self.i_stream = open(dir+'/'+filename + '.vm', 'r')
        self.parser = f_parser.Parser(self.i_stream)
        self.codeWriter = code_writer.codeWriter(self.dir, self.filename, self.parser)

    def Translate(self):
        while self.parser.hasMoreCommands():
            self.parser.advance()
            
            line_now = self.parser.line_now
            command_type = self.parser.commandType()

            if command_type == None:
                continue

            if command_type == 'C_PUSH':
                arg1 = self.parser.get_arg1
                arg2 = self.parser.get_arg2
                self.codeWriter.writePush(arg1, arg2)
            
            if command_type == 'C_POP':
                arg1 = self.parser.get_arg1
                arg2 = self.parser.get_arg2
                self.codeWriter.writePop(arg1, arg2)

            if command_type == 'C_ARITHMETRIC':
                command = self.parser.get_arg1()
                self.codeWriter.writeArithmetric(command)

        else:
            print('Finish VM translation.')

起きていること

始めに示したクラスのget_arg1,get_arg2メソッドでarg1とarg2を取得して処理しようとしましたが、以下のエラーが返ってきました。

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "~~~~\vm_translator.py", line 26, in Translate
    self.codeWriter.writePush(arg1, arg2)
  File "~~~~\code_writer.py", line 53, in writePush
    command = '\n'.join(address + process)
UnboundLocalError: local variable 'address' referenced before assignment

さらに、arg1とarg2の中身を確認すると

<bound method Parser.get_arg1 of <f_parser.Parser object at 0x0000023461595CD0>>

となっており、型としてstringを期待しているのにParserクラスのオブジェクトとして表示されています。エラー自体はget_argの戻り値がどのif文にも引っかからなかったことが原因です(さらにそのような場合のケアもしていない)。get_argメソッドの戻り値をarg1 == 'static'のような条件分岐に引っ掛けるにはどうすればよいでしょうか？クラス定義と継承あたりの基本事項と思われますが、どうぞ宜しくお願いします。

Comments

[LouiS0616]

writePush はどのように呼び出したのでしょうか？

[quickquip]

> def writePush(self, arg1, arg2):
がどこに書いてあるのか、

> 始めに示したクラスのget_arg1,get_arg2メソッドでarg1とarg2を取得して処理しようとしましたが
具体的になにをどう呼び出したのか、

とエラーの全文が載っていません。質問に追記しましょう。

Answer 1

arg1 = self.parser.get_arg1()
arg2 = self.parser.get_arg2()
self.codeWriter.writePush(arg1, arg2)

呼び出さないと返り値は返ってきません。

arg1 = self.parser.get_arg1
arg2 = self.parser.get_arg2
self.codeWriter.writePush(arg1, arg2)

Comments

[退会済みユーザー]

単純な見落としでした、、、。ご指摘ありがとうございます。