timesearch/timesearch_modules/getwiki.py

25 lines
764 B
Python
Raw Normal View History

import os
from . import common
from . import tsdb
def getwiki(subreddit):
(database, subreddit) = tsdb.TSDB.for_subreddit(subreddit, fix_name=True)
print('Getting wiki pages for /r/%s' % subreddit)
subreddit = common.r.subreddit(subreddit)
for wikipage in subreddit.wiki:
if wikipage.name == 'config/stylesheet':
continue
wikipage_path = database.wiki_dir.join(wikipage.name).replace_extension('md')
os.makedirs(wikipage_path.parent.absolute_path, exist_ok=True)
with open(wikipage_path.absolute_path, 'w', encoding='utf-8') as handle:
handle.write(wikipage.content_md)
print('Wrote', wikipage_path.relative_path)
def getwiki_argparse(args):
return getwiki(args.subreddit)