from seleniumbase import BaseCase class Tweet(object): def __init__(self, tweet=''): namerow = tweet + 'div.media-heading ' self.fullname = namerow + '.fullname' self.username = namerow + '.username' self.date = tweet + 'div.media-heading .heading-right' self.text = tweet + '.status-content-wrapper .status-content.media-body' self.retweet = tweet = '.retweet' class Profile(object): fullname = '.profile-card-fullname' username = '.profile-card-username' protected = '.protected-icon' verified = '.verified-icon' banner = '.profile-banner' bio = '.profile-bio' class BaseTestCase(BaseCase): def setUp(self): super(BaseTestCase, self).setUp() def tearDown(self): super(BaseTestCase, self).tearDown() def open_nitter(self, page=''): self.open(f'http://localhost:5000/{page}') def search_username(self, username): self.open_nitter() self.update_text('input', username) self.submit('form') def get_timeline_tweet(num=1): return Tweet(f'#tweets > div:nth-child({num}) ')