Add file name formatter class

bulkredditdownloader/file_name_formatter.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+# coding=utf-8
+
+import re
+from pathlib import Path
+
+import praw.models
+
+from bulkredditdownloader.resource import Resource
+
+
+class FileNameFormatter:
+    def __init__(self, file_format_string: str, directory_format_string: str):
+        self.file_format_string = file_format_string
+        self.directory_format_string = directory_format_string
+
+    @staticmethod
+    def _format_name(submission: praw.models.Submission, format_string: str) -> str:
+        submission_attributes = {
+            'title': submission.title,
+            'subreddit': submission.subreddit.display_name,
+            'redditor': submission.author.name,
+            'postid': submission.id,
+            'upvotes': submission.score,
+            'flair': submission.link_flair_text,
+            'date': submission.created_utc
+        }
+        result = format_string
+        for key in submission_attributes.keys():
+            if re.search(r'(?i).*{{{}}}.*'.format(key), result):
+                result = re.sub(r'(?i){{{}}}'.format(key), str(submission_attributes.get(key, 'unknown')), result)
+
+        result = result.replace('/', '')
+        return result
+
+    def format_path(self, resource: Resource, destination_directory: Path) -> Path:
+        subfolder = destination_directory / self._format_name(resource.source_submission, self.directory_format_string)
+        file_path = subfolder / (str(self._format_name(resource.source_submission,
+                                                       self.file_format_string)) + resource.extension)
+        return file_path