forked from s_ranjbar/city_retrofit
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
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)
|