Mercurial > code > home > repos > infra
diff templates/file-count/file_count.py @ 253:67fb7b27bfea
maildir file count service
author | drewp@bigasterisk.com |
---|---|
date | Tue, 09 Jan 2024 19:56:01 -0800 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/templates/file-count/file_count.py Tue Jan 09 19:56:01 2024 -0800 @@ -0,0 +1,20 @@ +import http.server +import socketserver +import os +import sys + +interface, port, dir, metric_name = sys.argv[1:] + + +class Web(http.server.SimpleHTTPRequestHandler): + + def do_GET(self): + files_count = len(os.listdir(dir)) + self.send_response(200) + self.send_header('Content-type', 'text/plain') + self.end_headers() + self.wfile.write(f'{metric_name} {files_count}'.encode()) + + +with socketserver.TCPServer((interface, int(port)), Web) as httpd: + httpd.serve_forever()