import codecs
import re

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

if 'refreshAdminRole' not in text:
    text = text.replace("import { hasAdminModuleAccess", "import { hasAdminModuleAccess, refreshAdminRole")
    
    # regex replace onMounted(() => { \s* initNavState()
    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("AdminLayout regex updated.")
else:
    print("Already there.")