class BackgroundManagerTest(unittest.TestCase):
def setUp(self):
self.background_manager = BackgroundManager()
def testGetWallpaper(self):
self.assertIn('file://',self.background_manager.getWallpaperUri())
def tearDown(self):
pass
class LastFmWallpaperUpdater():
def __init__(self, username):
self.username = username
self.wallpaper_path = ''
self.total_scrobblings = None
self.week_artists = None
self.modified_image = None
self.new_image_path = ""
def update_wallpaper(self):
if not self.wallpaper_path:
self.obtain_wallpaper_path()
self.obtain_lastfm_data()
self.generate_new_image()
self.generate_new_image_path()
self.modified_image.save(self.new_image_path,'JPEG')
self.set_new_wallpaper()
def obtain_wallpaper_path(self):
self.background_manager = BackgroundManager()
self.wallpaper_uri = self.background_manager.getWallpaperUri()
self.wallpaper_path = urlparse.urlparse(self.wallpaper_uri).path
def set_new_wallpaper(self):
self.background_manager = BackgroundManager()
self.wallpaper_uri = self.background_manager.setWallpaperUri('file://' + self.new_image_path)
def obtain_lastfm_data(self):
self.my_lastfm = LastFm(self.username)
self.total_scrobblings = self.my_lastfm.get_number_scrobblings()
self.week_artists = self.my_lastfm.get_week_artist(3)
def generate_new_image(self):
print 'Base image: ' + self.wallpaper_path
self.image_updater = ImageUpdater(self.wallpaper_path)
self.image_updater.add_username(self.username)
self.image_updater.add_scrobblings(self.total_scrobblings)
self.image_updater.weekly_chart(self.week_artists)
self.modified_image = self.image_updater.get_base_image()
def generate_new_image_path(self):
self.new_image_path = self.wallpaper_path[:(self.wallpaper_path.rfind('.')-1)]
self.new_image_path += '$'
self.new_image_path += self.wallpaper_path[self.wallpaper_path.rfind('.'):]