35 lines
910 B
Python
35 lines
910 B
Python
|
from os import listdir, path
|
||
|
|
|
||
|
|
def add_zeros(v: int, digits=3):
|
||
|
|
"""
|
||
|
|
return v as string, add leading zeros if len(str(v)) < digits
|
||
|
|
"""
|
||
|
|
s = str(v)
|
||
|
|
return '0' * (max(digits - len(s), 0)) + s
|
||
|
|
|
||
|
|
|
||
|
|
def get_next_digits(basename, directory=".", digits=3):
|
||
|
|
"""
|
||
|
|
get the next filename digits
|
||
|
|
example:
|
||
|
|
basename = file
|
||
|
|
directory has file001.csv, file002.pkl, file004.csv
|
||
|
|
-> return 005
|
||
|
|
"""
|
||
|
|
files = listdir(directory)
|
||
|
|
files.sort()
|
||
|
|
files.reverse()
|
||
|
|
lowest_number = -1
|
||
|
|
for file in files:
|
||
|
|
if not file.startswith(basename): continue
|
||
|
|
try:
|
||
|
|
dot = file.rfind('.')
|
||
|
|
if dot > 0: file = file[:dot]
|
||
|
|
number = int(file.replace(basename, ""))
|
||
|
|
if number < lowest_number: continue
|
||
|
|
lowest_number = number
|
||
|
|
except ValueError:
|
||
|
|
continue
|
||
|
|
|
||
|
|
return add_zeros(lowest_number+1)
|