36 lines
755 B
Python
Executable File
36 lines
755 B
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import argparse
|
|
import os
|
|
import socket
|
|
import subprocess
|
|
import time
|
|
|
|
from ltsdb_json import LTS
|
|
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("directory")
|
|
args = ap.parse_args()
|
|
|
|
# It's a bit weird that this works.
|
|
hostname = socket.gethostbyaddr(socket.gethostname())[0]
|
|
|
|
|
|
p = subprocess.run(["/usr/bin/dus", args.directory], capture_output=True, text=True)
|
|
t0 = time.time()
|
|
for ln in p.stdout.split("\n")[:-1]:
|
|
(size, directory) = ln.split("\t")
|
|
|
|
ts = LTS(
|
|
{
|
|
"hostname": hostname,
|
|
"directory": directory,
|
|
"measure": "diskusage",
|
|
"unit": "bytes",
|
|
}
|
|
)
|
|
ts.add(t0, int(size)*1024)
|
|
ts.save()
|
|
t1 = time.time()
|
|
print("recorded in", t1 - t0, "seconds")
|