docklet/tools/clean-usage.py

29 lines
730 B
Python
Executable File

#!/usr/bin/python3
import os, json, sys
sys.path.append("../src/")
from model import db, User, UserUsage
def clean_usage(username,alluser=False):
if alluser:
usages = UserUsage.query.all()
for usage in usages:
usage.cpu = str(0)
usage.memory = str(0)
usage.disk = str(0)
db.session.commit()
else:
usage = UserUsage.query.filter_by(username = username).first()
usage.cpu = str(0)
usage.memory = str(0)
usage.disk = str(0)
db.session.commit()
return
if __name__ == '__main__':
if len(sys.argv) >= 2:
username = sys.argv[1]
clean_usage(username)
else:
clean_usage("user",True)