From 10d25b3aa96d0baf25f369512c7f28ad86cd771a Mon Sep 17 00:00:00 2001
From: "minecraftchest1@outlook.com" <Minecraftchest1>
Date: Wed, 6 Nov 2024 10:57:44 -0600
Subject: [PATCH] Add A and AAAA records. Add invaliad data exception.

---
 main.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/main.py b/main.py
index 385c553..f7beab1 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,15 @@
 from enum import Enum
 from dataclasses import dataclass, field
 import time
+import ipaddress
+import dns
+
+class InvaliadDataException(Exception):
+	"""Exception raied when invaliad data is passed to a record"""
+
+	def __init__(self, message):
+		self.message	= message
+		super.__init__(self, message)
 
 @dataclass
 class Record:
@@ -19,6 +28,29 @@ class Record:
 	data: str	= '0.0.0.0'
 	ttl: int	= 6400
 
+@dataclass
+class A(Record):
+	def __init__(self, name: str = '@', ttl: str = 3600, data: str = '0.0.0.0'):
+		if isinstance(ipaddress.ip_address(data), ipaddress.IPv4Address):
+			self.data = data
+		else:
+			raise InvaliadDataException
+
+		self.rtype	= 'A'
+		self.name	= name
+		self.ttl	= ttl
+
+@dataclass
+class AAAA(Record):
+	def __init__(self, name: str = '@', ttl: str = 3600, data: str = '0.0.0.0'):
+		if isinstance(ipaddress.ip_address(data), ipaddress.IPv4Address):
+			self.data = data
+		else:
+			raise InvaliadDataException
+
+		self.rtype	= 'AAAA'
+		self.name	= name
+		self.ttl	= ttl
 
 
 @dataclass