/
rename_today.py
executable file
·123 lines (104 loc) · 4.95 KB
/
rename_today.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env python3
import argparse
p = argparse.ArgumentParser()
p.add_argument('files', nargs='*', help='files to rename, default to all files and dirs in current dir')
p.add_argument('-f', '-y', '--force', action='store_true', help='do not prompt, say yes to all questions')
g1 = p.add_mutually_exclusive_group()
g1.add_argument('--iso', action="store_true", help='use yyyy-mm-dd date format')
g1.add_argument('--iso-minutes', '--iso-time', action="store_true", help='use yyyy-mm-dd_HH-MM date format')
g1.add_argument('--iso-seconds', action="store_true", help='use yyyy-mm-dd_HH-MM-SS date format')
g1.add_argument('--long-weekday', action='store_true')
p.add_argument('--sep', default='-', help='separator used')
p.add_argument('--no-lowercase', action='store_true', help='Do not put in lowercase')
p.add_argument('--stat', action='store_true', help='do not take current time, look into the file "last time modified" field (returned by os.stat)')
p.add_argument('--no-check-dot-directory', action='store_true', help='do not take into account .directory file')
a = args = p.parse_args()
if a.iso or a.iso_minutes or a.iso_seconds or a.long_weekday:
assert a.sep == '-' and not a.no_lowercase
def parse_dot_directory(filename):
try:
with open(filename) as dot_directory:
lines = [line.strip('\n') for line in dot_directory]
Reg = Re('^\\[(.*)\\]$')
stops = [
i for i, line in enumerate(lines) if Reg.fullmatch(line)
] + [len(lines)]
for i in range(len(stops) - 1):
a,b = stops[i], stops[i+1]
name = Reg.fullmatch(lines[a]).group(1)
if name == 'rename_today.py':
for l in lines[a+1:b]:
a,b = l.split('=', maxsplit=1)
yield a,b
except FileNotFoundError:
return
import os
from re import compile as Re
if not args.files:
directory = '.'
if args.no_check_dot_directory:
dot_directory_infos = {}
else:
dot_directory_infos = dict(parse_dot_directory(os.path.join(directory, '.directory')))
args.file = list(filter(os.path.exists, os.listdir(directory)))
else:
dot_directory_infos = {}
args.file = args.files
assert all(map(os.path.exists, args.files)) # all(map(os.path.isfile OR os.path.directory, args.files))
date_format = (dot_directory_infos['strftime_datetime'] if 'strftime_datetime' in dot_directory_infos else
dot_directory_infos['strftime'] if 'strftime' in dot_directory_infos else
'%Y-%m-%d' if args.iso else
'%Y-%m-%d_%H-%M' if args.iso_minutes else
'%Y-%m-%d_%H-%M-%S' if args.iso_seconds else
'%A-%d-%b-%Y' if args.long_weekday else
'%a-%d-%b-%Y' if args.sep == '-' else
'%a-%d-%b-%Y'.replace('-', args.sep))
datetime_format = (dot_directory_infos['strftime_time'] if 'strftime_time' in dot_directory_infos else
dot_directory_infos['strftime'] if 'strftime' in dot_directory_infos else
'%Y-%m-%d_%H-%M-%S' if args.iso else
'%Y-%m-%d_%H-%M' if args.iso_minutes else
'%Y-%m-%d_%H-%M-%S' if args.iso_seconds else
'%A-%d-%b-%Y_%H:%M_%S' if args.long_weekday else
'%a-%d-%b-%Y_%H:%M_%S' if args.sep == '-' else
'%a-%d-%b-%Y_%H:%M_%S'.replace('-', args.sep))
try:
from file_renaming import Renamer
except ImportError:
class Renamer:
def __init__(self):
self.trycount = 0
def rename(self, a, b):
import os
if a == b:
return
if self.trycount >= 5 or input('Rename {!r} -> {!r}? [Y/n]'.format(a,b)).lower() not in ('n', 'no', 'non'):
os.rename(a, b)
else:
self.trycount += 1
class DummyRenamer:
def rename(self, a, b):
os.rename(a, b)
class DateFormatter:
def __init__(self, N):
self.N = N
def transform(self, s):
return s.lower() if not args.no_lowercase else s
def day_string_relative(self, n):
days = timedelta(days=1)
return self.transform((self.N + n * days).strftime(date_format))
def now_string(self):
return self.transform(self.N.strftime(datetime_format))
from datetime import datetime, timedelta
from itertools import chain
renamer = (Renamer() if not args.force else
DummyRenamer())
N = datetime.now()
for old in chain(
filter(os.path.isfile, args.file),
filter(os.path.isdir, args.file)): # dir after for weird file nesting like rename_today today/ today/today.txt
fmt = DateFormatter(N if not args.stat else datetime.fromtimestamp(os.stat(old).st_mtime))
new = (old.replace('now', fmt.now_string())
.replace('today', fmt.day_string_relative(0))
.replace('yesterday', fmt.day_string_relative(-1))
.replace('tomorrow', fmt.day_string_relative(1)))
renamer.rename(old, new)