43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
import re
|
||
|
|
|
||
|
|
def extract_userdata(file_path, vm_name, vmid):
|
||
|
|
with open(file_path, 'r') as f:
|
||
|
|
content = f.read()
|
||
|
|
|
||
|
|
# Find the section for this VM
|
||
|
|
# Pattern: VM: <name> (VMID: <id>) followed by separator, then userData until next separator
|
||
|
|
pattern = rf'VM: {re.escape(vm_name)}.*?VMID: {vmid}.*?==========================================\n(.*?)=========================================='
|
||
|
|
match = re.search(pattern, content, re.DOTALL)
|
||
|
|
|
||
|
|
if match:
|
||
|
|
userdata = match.group(1).strip()
|
||
|
|
# Remove any ANSI color codes if present
|
||
|
|
userdata = re.sub(r'\x1b\[[0-9;]*m', '', userdata)
|
||
|
|
return userdata
|
||
|
|
|
||
|
|
# Try alternative pattern without strict VM name match
|
||
|
|
pattern2 = rf'VM:.*?VMID: {vmid}.*?==========================================\n(.*?)=========================================='
|
||
|
|
match2 = re.search(pattern2, content, re.DOTALL)
|
||
|
|
if match2:
|
||
|
|
userdata = match2.group(1).strip()
|
||
|
|
userdata = re.sub(r'\x1b\[[0-9;]*m', '', userdata)
|
||
|
|
return userdata
|
||
|
|
|
||
|
|
return ""
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) != 4:
|
||
|
|
print("Usage: extract_userdata.py <file> <vm_name> <vmid>", file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
file_path = sys.argv[1]
|
||
|
|
vm_name = sys.argv[2]
|
||
|
|
vmid = sys.argv[3]
|
||
|
|
|
||
|
|
userdata = extract_userdata(file_path, vm_name, vmid)
|
||
|
|
if userdata:
|
||
|
|
print(userdata)
|
||
|
|
|