Add UUID syntax validation

This commit is contained in:
Peter Šurda 2021-12-02 15:43:43 +08:00
parent 95038a060c
commit a51b499cb3
Signed by: PeterSurda
GPG Key ID: 0C5F50C0B5F37D87
1 changed files with 9 additions and 1 deletions

10
main.py
View File

@ -7,6 +7,7 @@ import configparser
import os
import socket
import sys
import uuid as uuid_module
from ipaddress import AddressValueError, IPv4Address, IPv6Address
from os import access, R_OK
@ -33,10 +34,17 @@ class CloudInitRequest:
self.remoteip = None
self.hostinfo = ('localhost', )
self.request = request
self.uuid = uuid
self.meta_data = None
self.meta_data_loaded = False
self.user_data = None
try:
self.uuid = str(uuid_module.UUID('{' + uuid + '}'))
# ValueError is wrong UUID syntax
# TypeError is None
except (ValueError, TypeError):
self.uuid = None
self._init_ip()
self._generate_default_meta_data()