diff.py

import hashlib, logger, os, requests, subprocess, xattr 
import client as c
from mediafire.client import File

def diff(filename, mf_path=''):
  '''Use hashlib library to compare sha256 hash of
  current file to the sha256 hash stored on MediaFire'''

  full_expansion = get_path_expansion(filename)
  if(mf_path == '' or not mf_path):
    if ('/' in filename):
      mf_path = os.path.basename(filename)
    else:
      mf_path = filename

  if (os.path.isfile(full_expansion)):
    in_file = open(filename, 'r')
    mf_path = sanitize_path(mf_path)
    file_contents = in_file.read().encode('utf-8')
    new_hash = hashlib.sha256(file_contents).hexdigest()
    media_hash = get_hash(mf_path)
    try:
      old_hash = xattr.getxattr(full_expansion, 'hash').decode('ascii')
    except OSError:
      old_hash = '' 

    if (media_hash == ''):
      logger.die('No file path "' + mf_path + '/' + os.path.basename(filename) + '" in MediaFire')
    else:
      figure_time_scale(media_hash, old_hash, new_hash, os.path.basename(full_expansion))
  else:
    logger.die('No local file "' + os.path.basename(full_expansion) + '" found')

def get_path_expansion(path):
  return os.path.abspath(os.path.expanduser(path))

def get_hash(mf_path):
  try:
    client = c.get_client()  
    path = os.path.dirname(mf_path)
    if path == '/':
      path = ''
    contents = client.get_folder_contents_iter('mf:' + path + '/')
    for item in contents:
      if type(item) is File:
        if (item['filename'] == os.path.basename(mf_path)):
          return item['hash']
    return ''
  except requests.exceptions.RequestException:
    logger.die('Network error, please check network status and try again')

def figure_time_scale(media_hash, old_hash, new_hash, filename):
  if (old_hash == media_hash and new_hash != old_hash and new_hash != media_hash):
    logger.end('One: The local file, "' + filename + '", is ahead of the remote.')
  elif(old_hash == media_hash == new_hash):
    logger.end('One: All files are properly synced.')
  elif(old_hash == new_hash and media_hash != old_hash and media_hash != new_hash):
    logger.end('One: The remote file, "' + filename + '", is ahead of the local file.')
  elif(old_hash != new_hash != media_hash):
    logger.end('One: The remote and head are out of sync.')

def sanitize_path(path):
  if path:
    path = os.path.normpath(path)
    if (path[0] != '/'):
      path = '/' + path
    if (path[-1] == '/'):
      path = path[0:-1]
    return path
  return ''