How to implement a simple XML database in Python

Answer: Use Python's xml.etree.ElementTree module to build a lightweight XML database, suitable for storing configuration or small-scale data. 1. Design the XML structure, such as book information including title, author, year and other fields; 2. Use ET.parse() to load the XML file and parse it into a tree structure, and traverse the data through root.findall(); 3. Use ET.SubElement to create new elements when adding records and write them back to the file; 4. Search, update or delete data by finding matching items and modifying node content, and call tree.write() after each operation to persist and save. This solution is suitable for low concurrency scenarios, and professional databases such as SQLite should be used for large applications.

Building a simple XML database in Python is straightforward using the built-in xml.etree.ElementTree module. This approach lets you store, retrieve, and modify structured data using XML as the storage format—ideal for lightweight applications, configuration files, or small-scale data persistence.

1. Design Your XML Structure

Start by defining how your data will be organized. For example, if you're storing books, each book can be an element with child elements like title, author, and year.

Sample structure ( books.xml ):

<?xml version="1.0"?>
<library>
  <book id="1">
    <title>Python Basics</title>
    <author>John Doe</author>
    <year>2021</year>
  </book>
  <book id="2">
    <title>Advanced Python</title>
    <author>Jane Smith</author>
    <year>2023</year>
  </book>
</library>

2. Load and Parse the XML File

Use ElementTree to read and parse the XML file into a tree structure you can navigate.

import xml.etree.ElementTree as ET
<p>tree = ET.parse('books.xml')
root = tree.getroot()</p>

root now represents the top-level element. You can loop through all books:

for book in root.findall('book'):
    title = book.find('title').text
    author = book.find('author').text
    year = book.find('year').text
    print(f"{title} by {author}, {year}")

3.Add New Records

To insert a new book, create a new element and append it to the root:

def add_book(title, author, year):
    #Create new book element
    book = ET.SubElement(root, 'book')
    book.set('id', str(len(root) 1)) # Simple ID assignment
<pre class='brush:php;toolbar:false;'>ET.SubElement(book, &#39;title&#39;).text = title
ET.SubElement(book, &#39;author&#39;).text = author
ET.SubElement(book, &#39;year&#39;).text = str(year)

# Save back to file
tree.write(&#39;books.xml&#39;, encoding=&#39;utf-8&#39;, xml_declaration=True)

Call it like:

 add_book("Learning XML", "Alice Brown", 2024)

4. Search and Update Data

Find a book by title and update its year:

def update_book_year(title, new_year):
    for book in root.findall('book'):
        if book.find('title').text == title:
            book.find('year').text = str(new_year)
            tree.write('books.xml')
            return True
    return False

Delete a book by title:

def delete_book(title):
    for book in root.findall('book'):
        if book.find('title').text == title:
            root.remove(book)
            tree.write('books.xml')
            return True
    return False

These functions keep changes persistent by writing the updated tree back to the file.

Basically, this gives you a functional mini-database. It's not suitable for high-concurrency or large datasets, but it's perfect for local tools or prototypes. Use proper databases like SQLite for more robust needs.

The above is the detailed content of How to implement a simple XML database in Python. For more information, please follow other related articles on the PHP Chinese website!