Оптимизация кода

x = {'None' : ['global', set()]}
            
 
def add(x, namesp, arg):
    for k, v in x.items():
        if v[0] == namesp:
            tmp = k
            buf = v[1]
            break
    buf.add(arg)
    x[k] = [namesp, buf]    
 
 
def create(namesp, arg):
    return [namesp, set()]
 
 
def get(x, namesp, arg):
    if namesp == 'None':
        return
    for k, v in x.items():
        if v[0] == namesp:
            tmp = k
            if arg in v[1]:
                return namesp   
    return get(x, tmp, arg)
 
 
n = int(input())
for i in range(n):
    cmd, namesp, arg = input().split()
    if cmd == 'add':
        add(x, namesp, arg)
    elif cmd == 'create':
        x[arg] = create(namesp, arg)
    elif cmd == 'get':
        print(get(x, namesp, arg))

#!/usr/bin/python3
 
class Node(object):
    def __init__ (self, ns, parent=None):
        self.ns = ns
        self.parent = parent
        self.v = set()
 
 
class Tree(object):
    def __init__ (self):
        self.d = {'global': Node('global')}
 
    def create_ns(self, ns, parent):
        n = self.d[parent]
        self.d[ns] = Node(ns, n)
 
    def add_var(self, ns, var):
        n = self.d[ns]
        n.v.add(var)
 
    def find_var(self, ns, var):
        name = ns
        while True:
            n = self.d[name]
            if var in n.v:
                return n.ns
            if not n.parent:
                return None
            name = n.parent.ns
 
 
t = '''\
add global a
create foo global
add foo b
get foo a
get foo c
create bar foo
add bar a
get bar a
get bar b'''
 
tree = Tree()
 
for line in t.split('\n'):
    cmd, namesp, arg = line.split(' ')
    if cmd == 'create':
        tree.create_ns(namesp, arg)
    elif cmd == 'add':
        tree.add_var(namesp, arg)
    elif cmd == 'get':
        print(tree.find_var(namesp, arg))