Visualize Folder Structures with Python’s Treelib
Using Python's treelib library to output the contents of local directories as visual tree representations.
The first part of understanding any type of software is taking a glance at its file structure. It may seem like an outlandish and redundant statement to make to a generation who grew up on GUIs. GitHub is essentially no more than a GUI for Git, so it’s unsurprisingly that one of the largest company to follow a similar business model recently bought Github for millions.
All that said, a question remains: how do we being to understand closed source applications? If we can’t see the structure behind an app, I suppose we’ll have to build this model ourselves.
Treelib is a Python library that allows you to create a visual tree hierarchy: a simple plaintext representation of parent-child relationships.
Aside from scraping and mapping the intellectual property of others, Treelib comes in handy in situations where we have access to flat information (like a database table) where rows actually relate to one another (such as monolithic content-heavy site).
Treelib prints results like this:
It’s is a simple library, and only requires knowledge of a few lines of code in order to be used effectively. What’s more, we’re not simply spitting out flat useless data; we're storing these node relationships in memory. If needed, the trees we build can be modified or used for other the future.
Where da Treez At?
Install the Treelib package:
In your project, import Treelib:
Create a Tree with a Parent Node
The first step in utilizing Treelib is to create a tree object. We need to give our tree a name - this is essentially creating the top-level node that all other nodes will stem from.
In createNode(x, y), X is the value which will be displayed in the node, while Y is the unique identifier for that node. Children will be added to this parent node by referencing the unique identifier.
Note that in trees created with TreeLib, unique identifiers may only occur once. Therefore it is good to follow a sort of GUI system for identifying nodes.
Create Child Nodes
The last necessary part of creating a tree is, of course, populating the resulting children.
We will once again use create_node to add additional nodes, but these nodes will be associated with parents via parent=”x”. This will locate existing nodes in the tree by ID and associate these new nodes to that parent. This is why IDs must be unique for each node in the tree.
View the Tree
Finally, you'll want to view the fruits of your labor:
Way to go Johnny Appleseed, that’s pretty much the gist of it. There are additional features in the way Trees can be parse, and the way that nodes store additional data.
Check the official documentation for a full list of features.
If all you care about is printing the file structure of a current directory with zero interest in working with the actual data, you’re in luck (at least on Mac, hell if I know anything about Windows).
Unix systems come with a package named tree which does just what we want. On Mac OSX, we can install tree using Homebrew:
Go ahead and explore the various features of tree, such as writing to files or even doing so on a schedule. For now, here's some basic usage: