import dropbox
import dropbox.files
from NewsSocialSignaling import Helper
import datetime as dt


class DropboxUtils:
    dbx = dropbox.Dropbox(Helper.load_keys()['dropbox']['access_token'])
    print(dbx.users_get_current_account())

    def upload_infographic(self, graphic, username, name, share=False):
        path = '/infographics/{0}/{1}_{2}.png'.format(username, name, dt.datetime.utcnow().timestamp())
        self.dbx.files_upload(f=graphic, path=path, mode=dropbox.files.WriteMode.overwrite)

        # get shareable link
        link = None
        if share:
            link = self.dbx.sharing_create_shared_link(path=path, short_url=False, )
            link = link.url
            print(link)
        return path, link