Install the necessary packages by running

pip install -r requirements.txt

This file contains HOST and PORT values. Set it to 127.0.0.1 and 53 (since default DNS port is 53)

Creates <domain>.json JSON list of DNS records for a particular domain of a particular record type.
Edit the recordtype and domain and then run the file.
Copy the contents of this file to zones/dns.json

Start the DNS server by running

sudo python dns.py

The sudo command is required since the program's UDP socket will be running on the reserved port 53.

Open the terminal and run the dig command
dig <domain> @127.0.0.1
dig <record-type> <domain> @127.0.0.1

Example: dig google.com @127.0.0.1
Example: dig MX google.com @127.0.0.1

NOTE: Currently only A type records are supported.

[ ] Replace zones/dns.json with a Zone File
[ ] Add support for other record types (MX, CNAME, etc)