| 1234567891011121314151617181920212223242526272829303132 |
- import os
- import requests
- from . import common
- from . import tsdb
- session = requests.Session()
- def get_styles(subreddit):
- (database, subreddit) = tsdb.TSDB.for_subreddit(subreddit, fix_name=True)
- print('Getting styles for /r/%s' % subreddit)
- subreddit = common.r.subreddit(subreddit)
- styles = subreddit.stylesheet()
- database.styles_dir.makedirs(exist_ok=True)
- stylesheet_filepath = database.styles_dir.with_child('stylesheet.css')
- print('Downloading %s' % stylesheet_filepath.relative_path)
- with stylesheet_filepath.open('w', encoding='utf-8') as stylesheet:
- stylesheet.write(styles.stylesheet)
- for image in styles.images:
- image_basename = image['name'] + '.' + image['url'].split('.')[-1]
- image_filepath = database.styles_dir.with_child(image_basename)
- print('Downloading %s' % image_filepath.relative_path)
- with image_filepath.open('wb') as image_file:
- response = session.get(image['url'])
- image_file.write(response.content)
- def get_styles_argparse(args):
- return get_styles(args.subreddit)
|
