2021-02-07 17:46:20 +13:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
2020-05-29 06:42:11 +12:00
|
|
|
import io
|
2021-02-07 17:46:20 +13:00
|
|
|
import logging
|
2021-02-07 01:29:13 +13:00
|
|
|
import pathlib
|
2020-05-29 06:42:11 +12:00
|
|
|
from pathlib import Path
|
|
|
|
|
2021-02-07 20:08:24 +13:00
|
|
|
from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
|
2021-02-07 14:05:18 +13:00
|
|
|
from bulkredditdownloader.errors import FileAlreadyExistsError, TypeInSkip
|
|
|
|
from bulkredditdownloader.utils import GLOBAL
|
2020-05-29 06:42:11 +12:00
|
|
|
|
2021-02-07 17:46:20 +13:00
|
|
|
logger = logging.getLogger(__name__)
|
2021-02-25 22:32:06 +13:00
|
|
|
|
2020-05-29 06:42:11 +12:00
|
|
|
|
2021-02-07 14:33:19 +13:00
|
|
|
class SelfPost(BaseDownloader):
|
2021-02-07 01:29:13 +13:00
|
|
|
def __init__(self, directory: pathlib.Path, post: dict):
|
2021-02-07 14:33:19 +13:00
|
|
|
super().__init__(directory, post)
|
2021-02-07 17:46:20 +13:00
|
|
|
self.download()
|
|
|
|
|
|
|
|
def download(self):
|
2021-02-25 22:32:06 +13:00
|
|
|
if "self" in GLOBAL.arguments.skip:
|
|
|
|
raise TypeInSkip
|
2020-06-04 03:10:25 +12:00
|
|
|
|
2021-02-07 17:46:20 +13:00
|
|
|
self.directory.mkdir(exist_ok=True)
|
|
|
|
filename = GLOBAL.config['filename'].format(**self.post)
|
2020-05-29 06:42:11 +12:00
|
|
|
|
2021-02-07 17:46:20 +13:00
|
|
|
file_dir = self.directory / (filename + ".md")
|
|
|
|
logger.info(file_dir)
|
|
|
|
logger.info(filename + ".md")
|
2020-05-29 06:42:11 +12:00
|
|
|
|
2021-02-06 21:35:50 +13:00
|
|
|
if Path.is_file(file_dir):
|
2020-05-29 06:42:11 +12:00
|
|
|
raise FileAlreadyExistsError
|
2021-02-25 22:32:06 +13:00
|
|
|
|
2020-05-29 06:42:11 +12:00
|
|
|
try:
|
2021-02-07 17:46:20 +13:00
|
|
|
self._write_to_file(file_dir, self.post)
|
2020-05-29 06:42:11 +12:00
|
|
|
except FileNotFoundError:
|
2021-02-07 17:46:20 +13:00
|
|
|
file_dir = self.post['POSTID'] + ".md"
|
|
|
|
file_dir = self.directory / file_dir
|
2020-05-29 06:42:11 +12:00
|
|
|
|
2021-02-07 17:46:20 +13:00
|
|
|
self._write_to_file(file_dir, self.post)
|
2021-02-25 22:32:06 +13:00
|
|
|
|
2020-05-29 06:42:11 +12:00
|
|
|
@staticmethod
|
2021-02-07 17:46:20 +13:00
|
|
|
def _write_to_file(directory: pathlib.Path, post: dict):
|
2020-05-29 06:42:11 +12:00
|
|
|
"""Self posts are formatted here"""
|
|
|
|
content = ("## ["
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["TITLE"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ "]("
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["CONTENTURL"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ ")\n"
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["CONTENT"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ "\n\n---\n\n"
|
|
|
|
+ "submitted to [r/"
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["SUBREDDIT"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ "](https://www.reddit.com/r/"
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["SUBREDDIT"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ ") by [u/"
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["REDDITOR"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ "](https://www.reddit.com/user/"
|
2020-06-02 00:05:02 +12:00
|
|
|
+ post["REDDITOR"]
|
2020-05-29 06:42:11 +12:00
|
|
|
+ ")")
|
|
|
|
|
2021-02-25 22:32:06 +13:00
|
|
|
with io.open(directory, "w", encoding="utf-8") as FILE:
|
2021-02-07 17:46:20 +13:00
|
|
|
print(content, file=FILE)
|
|
|
|
logger.info("Downloaded")
|