Skip to content

Commit

Permalink
Feature added: Issue #63
Browse files Browse the repository at this point in the history
  • Loading branch information
AamerShareef authored and AamerShareef committed Jul 1, 2023
1 parent f94aa2e commit 870eb6a
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 6 deletions.
26 changes: 25 additions & 1 deletion phockup.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,6 +287,28 @@ def parse_args(args=sys.argv[1:]):
"""
)

parser.add_argument(
'--filename_suffix',
type=str,
default='',
help="""\
String to append to the output file name. if --filename_add_original is set, this will be added after the original filename.
This string can also be setting of the output name (e.g. via $USER,
$HOSTNAME, %USERNAME%, etc.)
"""

)


parser.add_argument(
'-ao',
'--filename_add_original',
action='store_true',
help="""\
Add original filename to the target filename
""",
)

parser.add_argument(
'--output_suffix',
type=str,
Expand Down Expand Up @@ -344,7 +366,9 @@ def main(options):
no_date_dir=options.no_date_dir,
skip_unknown=options.skip_unknown,
output_prefix=options.output_prefix,
output_suffix=options.output_suffix
output_suffix=options.output_suffix,
filename_suffix=options.filename_suffix,
filename_add_original=options.filename_add_original
)


Expand Down
30 changes: 25 additions & 5 deletions src/phockup.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,10 @@ def __init__(self, input_dir, output_dir, **args):
self.output_dir = output_dir
self.output_prefix = args.get('output_prefix' or None)
self.output_suffix = args.get('output_suffix' or '')

self.filename_suffix=args.get('filename_suffix' or '')
self.filename_add_original = args.get('filename_add_original', False)

self.no_date_dir = args.get('no_date_dir') or Phockup.DEFAULT_NO_DATE_DIRECTORY
self.dir_format = args.get('dir_format') or os.path.sep.join(Phockup.DEFAULT_DIR_FORMAT)
self.move = args.get('move', False)
Expand Down Expand Up @@ -203,7 +207,7 @@ def get_file_name(self, original_filename, date):
"""
if self.original_filenames:
return os.path.basename(original_filename)

try:
filename = [
f'{date["date"].year :04d}',
Expand All @@ -217,8 +221,23 @@ def get_file_name(self, original_filename, date):

if date['subseconds']:
filename.append(date['subseconds'])

return ''.join(filename) + os.path.splitext(original_filename)[1]

# target filename with timestamp as name
target_filename=''.join(filename)

if self.filename_add_original:
base_name=os.path.basename(original_filename)
f_name=os.path.splitext(base_name)[0]
target_filename=''.join(filename) + '_' + f_name
#debug
# print(target_filename+os.path.splitext(original_filename)[1])

if self.filename_suffix:
target_filename=target_filename + '_' + self.filename_suffix
# print(target_filename)

return target_filename + os.path.splitext(original_filename)[1]

# TODO: Double check if this is correct!
except TypeError:
return os.path.basename(original_filename)
Expand Down Expand Up @@ -335,8 +354,9 @@ def get_file_name_and_path(self, filename):
self.date_field)
output = self.get_output_dir(date)
target_file_name = self.get_file_name(filename, date)
if not self.original_filenames:
target_file_name = target_file_name.lower()

# if not self.original_filenames:
# target_file_name = target_file_name.lower()
else:
output = self.get_output_dir([])
target_file_name = os.path.basename(filename)
Expand Down

0 comments on commit 870eb6a

Please sign in to comment.