docker.images/ansible.awx/awx-17.1.0/awx/plugins/isolated/awx_isolated_cleanup.py

75 lines
2.8 KiB
Python

# Copyright (c) 2017 Ansible by Red Hat
#
# This file is part of Ansible Tower, but depends on code imported from Ansible.
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
from ansible.module_utils.basic import AnsibleModule
import glob
import os
import shutil
import datetime
import subprocess
def main():
module = AnsibleModule(
argument_spec = dict()
)
changed = False
paths_removed = set([])
# If a folder was last modified before this datetime, it will always be deleted
folder_cutoff = datetime.datetime.now() - datetime.timedelta(days=7)
# If a folder does not have an associated job running and is older than
# this datetime, then it will be deleted because its job has finished
job_cutoff = datetime.datetime.now() - datetime.timedelta(hours=1)
BASE_DIR = '/tmp'
bwrap_pattern = 'bwrap_[0-9]*_*'
private_data_dir_pattern = 'awx_[0-9]*_*'
bwrap_path_pattern = os.path.join(BASE_DIR, bwrap_pattern)
for bwrap_path in glob.iglob(bwrap_path_pattern):
st = os.stat(bwrap_path)
modtime = datetime.datetime.fromtimestamp(st.st_mtime)
if modtime > job_cutoff:
continue
elif modtime > folder_cutoff:
private_data_dir_path_pattern = os.path.join(BASE_DIR, bwrap_path, private_data_dir_pattern)
private_data_dir_path = next(glob.iglob(private_data_dir_path_pattern), None)
if private_data_dir_path:
try:
if subprocess.check_call(['ansible-runner', 'is-alive', private_data_dir_path]) == 0:
continue
except subprocess.CalledProcessError:
# the job isn't running anymore, clean up this path
module.debug('Deleting path {} its job has completed.'.format(bwrap_path))
module.debug('Deleting path {} due to private_data_dir not being found.'.format(bwrap_path))
else:
module.debug('Deleting path {} because modification date is too old.'.format(bwrap_path))
changed = True
paths_removed.add(bwrap_path)
shutil.rmtree(bwrap_path)
module.exit_json(changed=changed, paths_removed=list(paths_removed))
if __name__ == '__main__':
main()