import codecs
import re

path = r'd:\骅锋\mall\layouts\admin\AdminLayout.uvue'
with codecs.open(path, 'r', 'utf-8') as f:
    text = f.read()

# Make sure we add the await
if 'await refreshAdminRole()' not in text:
    text = re.sub(r'onMounted\(\(\) => \{\s+initNavState\(\)',
        "onMounted(async () => {\n  await refreshAdminRole()\n  initNavState()",
        text)
    
    with codecs.open(path, 'w', 'utf-8') as f:
        f.write(text)
    print("Injected await refreshAdminRole()")
else:
    print("Await already present.")