122 lines
2.9 KiB
Python
Executable File
122 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import os, argparse, sys
|
|
from shutil import copyfile, copytree
|
|
from datetime import datetime
|
|
|
|
def get_options():
|
|
parser = argparse.ArgumentParser(
|
|
description = 'Rename file with version number'
|
|
)
|
|
parser.add_argument(
|
|
'-c',
|
|
action = "store_false",
|
|
dest = "move",
|
|
default = True,
|
|
help = "Copy path instead of moving",
|
|
)
|
|
parser.add_argument(
|
|
'-d',
|
|
action = "store_true",
|
|
dest = "date",
|
|
help = "Use paths date+time as version number",
|
|
default = False
|
|
)
|
|
parser.add_argument(
|
|
'-f',
|
|
action = "store_true",
|
|
dest = "force",
|
|
help = "Force overwrite",
|
|
default = False
|
|
)
|
|
parser.add_argument(
|
|
'-q',
|
|
action = "store_true",
|
|
dest = "quiet",
|
|
help = "Quiet operation",
|
|
default = False
|
|
)
|
|
parser.add_argument(action = "store", dest = "file")
|
|
return parser.parse_args()
|
|
|
|
|
|
def get_version_name(full_path):
|
|
""" New name versioned with integer """
|
|
file_dir = os.path.dirname(full_path)
|
|
file_name = os.path.basename(full_path)
|
|
basename, extension = os.path.splitext(file_name)
|
|
version = 1
|
|
|
|
while True:
|
|
new_name = os.path.join(
|
|
file_dir,
|
|
"%s.v%02d%s"%(
|
|
basename,
|
|
version,
|
|
extension
|
|
)
|
|
)
|
|
if os.path.exists(new_name):
|
|
version += 1
|
|
else:
|
|
break
|
|
return new_name
|
|
|
|
|
|
def get_date_name(full_path):
|
|
""" New name versioned with date of the file """
|
|
file_dir = os.path.dirname(full_path)
|
|
file_name = os.path.basename(full_path)
|
|
basename, extension = os.path.splitext(file_name)
|
|
file_time = os.stat(full_path).st_mtime
|
|
time_formatted = datetime.fromtimestamp(
|
|
file_time
|
|
).strftime("%Y%m%d_%H%M%S")
|
|
|
|
new_name = os.path.join(
|
|
file_dir,
|
|
"%s.%s%s"%(
|
|
basename,
|
|
time_formatted,
|
|
extension
|
|
)
|
|
)
|
|
|
|
if full_path.endswith("%s%s"%(
|
|
time_formatted,
|
|
extension
|
|
)):
|
|
# file is already named with the date+time
|
|
return full_path
|
|
return new_name
|
|
|
|
|
|
def test_existing(name, overwrite):
|
|
if os.path.exists(name):
|
|
if overwrite:
|
|
os.remove(name)
|
|
else:
|
|
sys.stderr.write(
|
|
"'%s' already exists, exiting.\n"%(
|
|
name
|
|
)
|
|
)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
opts = get_options()
|
|
if opts.date:
|
|
new_name = get_date_name(opts.file)
|
|
else:
|
|
new_name = get_version_name(opts.file)
|
|
if not opts.quiet:
|
|
print(new_name)
|
|
test_existing(new_name, opts.force)
|
|
if opts.move:
|
|
os.rename(opts.file, new_name)
|
|
else:
|
|
if os.path.isdir(opts.file):
|
|
copytree(opts.file, new_name)
|
|
else:
|
|
copyfile(opts.file, new_name)
|