1
0
mirror of https://github.com/instaloader/instaloader.git synced 2024-10-03 22:07:11 +02:00

InstaloaderContext class for log and low-level IO

This commit is contained in:
Alexander Graf 2018-04-05 00:20:03 +02:00
parent 8623829696
commit c350847b50
2 changed files with 504 additions and 463 deletions

File diff suppressed because it is too large Load Diff

View File

@ -25,10 +25,10 @@ class TestInstaloader(unittest.TestCase):
self.L = instaloader.Instaloader(download_geotags=instaloader.Tristate.always,
download_comments=instaloader.Tristate.always,
save_metadata=instaloader.Tristate.always)
self.L.raise_all_errors = True
self.L.context.raise_all_errors = True
def tearDown(self):
self.L.session.close()
self.L.close()
os.chdir('/')
print("Removing {}".format(self.dir))
shutil.rmtree(self.dir)
@ -40,14 +40,12 @@ class TestInstaloader(unittest.TestCase):
@unittest.SkipTest
def test_stories_download(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
self.L.download_stories()
@unittest.SkipTest
def test_private_profile_download(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
self.L.download_profile(PRIVATE_PROFILE, download_stories=True)
def test_profile_pic_download(self):
@ -57,45 +55,38 @@ class TestInstaloader(unittest.TestCase):
self.L.download_hashtag(HASHTAG, NORMAL_MAX_COUNT)
def test_feed_download(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
self.L.download_feed_posts(NORMAL_MAX_COUNT)
def test_feed_paging(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
for count, post in enumerate(self.L.get_feed_posts()):
print(post)
if count == PAGING_MAX_COUNT:
break
def test_saved_download(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
self.L.download_saved_posts(NORMAL_MAX_COUNT)
def test_saved_paging(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
for count, post in enumerate(instaloader.Profile(self.L, OWN_USERNAME).get_saved_posts()):
self.L.load_session_from_file(OWN_USERNAME)
for count, post in enumerate(instaloader.Profile(self.L.context, OWN_USERNAME).get_saved_posts()):
print(post)
if count == PAGING_MAX_COUNT:
break
def test_test_login(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
self.assertEqual(OWN_USERNAME, self.L.test_login())
def test_get_followees(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
for f in self.L.get_followees(OWN_USERNAME):
print(f['username'])
def test_get_followers(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
self.L.load_session_from_file(OWN_USERNAME)
for f in self.L.get_followers(OWN_USERNAME):
print(f['username'])
@ -106,16 +97,15 @@ class TestInstaloader(unittest.TestCase):
self.assertEqual(PUBLIC_PROFILE_ID, self.L.get_id_by_username(PUBLIC_PROFILE))
def test_get_likes(self):
if not self.L.is_logged_in:
self.L.load_session_from_file(OWN_USERNAME)
for post in instaloader.Profile(self.L, OWN_USERNAME).get_posts():
self.L.load_session_from_file(OWN_USERNAME)
for post in instaloader.Profile(self.L.context, OWN_USERNAME).get_posts():
for like in post.get_likes():
print(like['username'])
break
def test_post_from_mediaid(self):
for post in instaloader.Profile(self.L, PUBLIC_PROFILE).get_posts():
post2 = instaloader.Post.from_mediaid(self.L, post.mediaid)
for post in instaloader.Profile(self.L.context, PUBLIC_PROFILE).get_posts():
post2 = instaloader.Post.from_mediaid(self.L.context, post.mediaid)
self.assertEqual(post, post2)
break