mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 11:36:54 +00:00
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from distutils import log
|
|
from distutils.errors import DistutilsOptionError
|
|
from distutils.util import convert_path
|
|
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)
|