Написать функцию , которая выполняет сложение и вычитание данной строки

def arf(s):
    return eval(s)
 
def main():
    print(arf("3+6-1"))
        
if __name__ == "__main__":
    main()

import re
 
pattern = re.compile(r"([+-])")
 
def arf(s):
    tokens = pattern.split(s)
    while len(tokens) >= 3:
        first, operator, second = (tokens.pop(0) for i in range(3))
        first, second = int(first), int(second)
        result = None
        if operator == "+":
            result = first + second
        elif operator == "-":
            result = first - second
        else:
            raise ValueError("Wrong token: '{}'".format(operator))
        tokens.insert(0, result)
    result = tokens.pop()
    if tokens:
        raise ValueError("Unproccessed tokens in stack: {}".format(tokens))
    return result
 
def main():
    print(arf("3+6-1"))
        
if __name__ == "__main__":
    main()