diff --git a/.gitignore b/.gitignore index 9facd35..7a045df 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ .pip *.jpg +images/ diff --git a/images/.stanis-tits.latest b/images/.stanis-tits.latest deleted file mode 100644 index c227083..0000000 --- a/images/.stanis-tits.latest +++ /dev/null @@ -1 +0,0 @@ -0 \ No newline at end of file diff --git a/stanis-tits.py b/stanis-tits.py index bff7fd8..837b876 100644 --- a/stanis-tits.py +++ b/stanis-tits.py @@ -1,34 +1,42 @@ #!/usr/bin/env python3 -import sys -sys.path.append('./.pip') -import requests -from bs4 import BeautifulSoup import re import shutil import os.path +try: + import requests + from bs4 import BeautifulSoup +except ImportError: + import sys + sys.path.append('./.pip') + import requests + from bs4 import BeautifulSoup __author__ = 'Alexander Popov' -__version__ = '0.0.1' +__version__ = '0.1.0' __license__ = 'Unlicense' IMAGES_DIR = './images' COOKIES = dict(block='951') URL = 'http://blog.stanis.ru/?back=%d' PAGE = 0 +if not os.path.exists('%s/.stanis-tits.latest' % IMAGES_DIR): + if not os.path.exists('%s' % IMAGES_DIR): + os.mkdir('%s' % IMAGES_DIR) + + with open('%s/.stanis-tits.latest' % IMAGES_DIR, 'w') as f: + f.write('0') with open('%s/.stanis-tits.latest' % IMAGES_DIR, 'r') as f: LATEST_FILE = f.read() STOP = False NEXT_LATEST = None -while STOP == False: +while STOP is False: print('Loading page %d' % PAGE) r = requests.get(URL % PAGE, cookies=COOKIES) - - soup = BeautifulSoup(r.text.encode('cp1251'), - "html.parser", from_encoding="windows-1251") + "html.parser", from_encoding="windows-1251") images = soup.findAll('img', src=re.compile('img/*')) for image in images: @@ -36,17 +44,18 @@ while STOP == False: STOP = True if PAGE == 0: - if NEXT_LATEST == None: + if NEXT_LATEST is None: NEXT_LATEST = str(image['src'].split('/')[1].split('.')[0]) with open('%s/.stanis-tits.latest' % IMAGES_DIR, 'w+') as f: f.write(NEXT_LATEST) - if not os.path.exists('%s/%s' % (IMAGES_DIR, image['src'].split('/')[1],)): + if not os.path.exists('%s/%s' % (IMAGES_DIR, + image['src'].split('/')[1],)): print('\tDownload %s' % image['src'].split('/')[1]) - response = requests.get('http://blog.stanis.ru/%s' % image['src'], stream=True) - - with open('%s/%s' % (IMAGES_DIR, image['src'].split('/')[1]), 'wb') as out_image: + response = requests.get('http://blog.stanis.ru/%s' + % image['src'], stream=True) + with open('%s/%s' % (IMAGES_DIR, image['src'].split('/')[1]), + 'wb') as out_image: shutil.copyfileobj(response.raw, out_image,) PAGE += 1 -