munyal-client/dir_to_json.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import os
from hashlib import md5
from misc import path_join


def md5sum(filename):
    try:
        hash = md5()
        with open(filename, "rb") as f:
            for chunk in iter(lambda: f.read(128 * hash.block_size), b""):
                hash.update(chunk)
        return hash.hexdigest()
    except Exception:
        return None


def get_json(path):
    out = []
    items = os.listdir(path)
    try:
        for item in items:
            if item[0] != "." and not item.endswith("tmp"):
                item_json = {"name": item}
                route = path_join(path, item)
                if os.path.isdir(route):
                    item_json["is_file"] = False
                    item_json["content"] = get_json(route)
                elif os.path.isfile(route):
                    item_json["is_file"] = True
                    item_json["size"] = os.path.getsize(route)
                    item_json["last_modified"] = os.path.getmtime(route)
                    item_json["created_at"] = os.path.getctime(route)
                    checksum = md5sum(route)
                    if checksum:
                        item_json["checksum"] = checksum
                    else:
                        item = None
                else:
                    continue
                out.append(item_json)
    except Exception:
        return get_json(path)
    return out


if __name__ == "__main__":
    import sys
    output = get_json(sys.argv[1])
    print(json.dumps(output, indent=4))