Spaces:
Sleeping
Sleeping
| from distutils.util import convert_path | |
| from distutils import log | |
| from distutils.errors import DistutilsOptionError | |
| import os | |
| import shutil | |
| from setuptools import Command | |
| class rotate(Command): | |
| """Delete older distributions""" | |
| description = "delete older distributions, keeping N newest files" | |
| user_options = [ | |
| ('match=', 'm', "patterns to match (required)"), | |
| ('dist-dir=', 'd', "directory where the distributions are"), | |
| ('keep=', 'k', "number of matching distributions to keep"), | |
| ] | |
| boolean_options = [] | |
| def initialize_options(self): | |
| self.match = None | |
| self.dist_dir = None | |
| self.keep = None | |
| def finalize_options(self): | |
| if self.match is None: | |
| raise DistutilsOptionError( | |
| "Must specify one or more (comma-separated) match patterns " | |
| "(e.g. '.zip' or '.egg')" | |
| ) | |
| if self.keep is None: | |
| raise DistutilsOptionError("Must specify number of files to keep") | |
| try: | |
| self.keep = int(self.keep) | |
| except ValueError as e: | |
| raise DistutilsOptionError("--keep must be an integer") from e | |
| if isinstance(self.match, str): | |
| self.match = [ | |
| convert_path(p.strip()) for p in self.match.split(',') | |
| ] | |
| self.set_undefined_options('bdist', ('dist_dir', 'dist_dir')) | |
| def run(self): | |
| self.run_command("egg_info") | |
| from glob import glob | |
| for pattern in self.match: | |
| pattern = self.distribution.get_name() + '*' + pattern | |
| files = glob(os.path.join(self.dist_dir, pattern)) | |
| files = [(os.path.getmtime(f), f) for f in files] | |
| files.sort() | |
| files.reverse() | |
| log.info("%d file(s) matching %s", len(files), pattern) | |
| files = files[self.keep:] | |
| for (t, f) in files: | |
| log.info("Deleting %s", f) | |
| if not self.dry_run: | |
| if os.path.isdir(f): | |
| shutil.rmtree(f) | |
| else: | |
| os.unlink(f) | |