Source code for openquake.commands.purge

# -*- coding: utf-8 -*-
# vim: tabstop=4 shiftwidth=4 softtabstop=4
# Copyright (C) 2015-2020 GEM Foundation
# OpenQuake is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# OpenQuake is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# GNU Affero General Public License for more details.
# You should have received a copy of the GNU Affero General Public License
# along with OpenQuake. If not, see <>.
import os
import re
import getpass
from openquake.baselib import sap, datastore
from openquake.commonlib.logs import dbcmd

datadir = datastore.get_datadir()

[docs]def purge_one(calc_id, user, force): """ Remove one calculation ID from the database and remove its datastore """ dbcmd('del_calc', calc_id, user, force) f1 = os.path.join(datadir, 'calc_%s.hdf5' % calc_id) f2 = os.path.join(datadir, 'calc_%s_tmp.hdf5' % calc_id) for f in [f1, f2]: if os.path.exists(f): # not removed yet os.remove(f) print('Removed %s' % f)
# used in the reset command
[docs]def purge_all(user=None): """ Remove all calculations of the given user """ user = user or getpass.getuser() if os.path.exists(datadir): for fname in os.listdir(datadir): mo = re.match(r'(calc_|cache_)(\d+)\.hdf5', fname) if mo is not None: calc_id = int( purge_one(calc_id, user, force=True)
[docs]@sap.script def purge(calc_id, force=False): """ Remove the given calculation. If you want to remove all calculations, use oq reset. """ if calc_id < 0: try: calc_id = datastore.get_calc_ids(datadir)[calc_id] except IndexError: print('Calculation %d not found' % calc_id) return purge_one(calc_id, getpass.getuser(), force)
purge.arg('calc_id', 'calculation ID', type=int) purge.flg('force', 'ignore dependent calculations')