sfs

#!/usr/bin/env python
"""Snapchat FS

Usage:
  sfs list
  sfs download_all <target_directory>
  sfs download <target_directory> <snap_id> ...
  sfs upload <file> ...
  sfs (-h | --help)
  sfs (-v | --version)

Options:
  -h --help        Show this screen
  -v --version     Show the version
"""
from docopt import docopt
import os, getpass, sys
from snapchat_core import constants, SfsSession
from snapchat_fs import *

__author__ = "Alex Clemmer, Chad Brubaker"
__copyright__ = "Copyright 2013, Alex Clemmer and Chad Brubaker"
__credits__ = ["Alex Clemmer", "Chad Brubaker"]

__license__ = "MIT"
__version__ = "0.1"
__maintainer__ = "Alex Clemmer"
__email__ = "clemmer.alexander@gmail.com"
__status__ = "Prototype"


# SETTINGS CONSTANTS -- possible settings found in config file ~/.snapchat_fs
USERNAME = 'USERNAME'
PASSWORD = 'PASSWORD'

# SETTINGS DICTIONARY -- stores defaults found in ~/.snapchat_fs
#     contains dummy entries for every possible setting so that we can
#     validate whether a setting found in config file is valid based on
#     whether it appears in this dictionary
SETTINGS = {
      USERNAME: ""
    , PASSWORD: ""
}


def initialize_user():
    """
    Initializes user if we can't find the config file (which is
    ~/.snapchat_fs).

    This usually involves asking for username and password, building the
    config file, and writing these values to the file.
    """
    # Welcome message
    print "Looks like you haven't used Snapchat FS before."
    print "We're going to ask for a username and a password; you can"
    print "change these later by messing with the file %s" % \
            constants.CONFIG_FILE_PATH

    # get username/password
    print "Please enter a username:"
    username = raw_input()
    print "Please enter your password (NOTE: STORED IN PLAINTEXT):"
    password = getpass.getpass('')

    # Try logging in first before we continue
    ss = SfsSession(username, password)
    ss.login()
    # write username/password to file
    with open(constants.CONFIG_FILE_PATH, 'w') as w:
        # Store the username in lowercase for consistency.
        # Snapchat is case-insensitive with both usernames and emails.
        w.write("USERNAME=%s\n" % username.lower())
        w.write("PASSWORD=%s\n" % password)

def init_settings_from_config_file():
    """
    Create settings dictionary mapping attribute -> value. Contains, e.g.,
    username and password.

    The attributes are a list of constants at the top of the file, e.g.,
    `USERNAME` and `PASSWORD`.
    @return Settings dictionary.
    """
    # create config file with username/password if it doesn't exist
    if not os.path.exists(constants.CONFIG_FILE_PATH):
        # can fail and raise a login exception, caught by main()
        initialize_user()

    # open config file, populate settings dictionary with fields we find
    with open(constants.CONFIG_FILE_PATH) as f:
        for line in f.readlines():
            split_line = line.split("=")
            if len(split_line) != 2:
                raise Exception("\033[1;31mMalformed line in config file %s\033[0m" \
                                % line)
            field = split_line[0].strip()
            value = split_line[1].strip()

            # validate that dummy entry exists; if it doesn't, then it's
            # not a valid setting
            if field in SETTINGS:
                SETTINGS[field] = value
            else:
                raise Exception(
                        "\033[1;31mUnknown setting in config file %s\033[0m" \
                        % constants.CONFIG_FILE_PATH)
        if SETTINGS[USERNAME] == '' or SETTINGS[PASSWORD] == '':
            print "\033[1;31mUsername or password missing from config file %s\033[0m" \
                    % constants.CONFIG_FILE_PATH

    return SETTINGS

def create_dir_if_not_exists(directory):
    """
    Simple utility function creates a directory if it doesn't exist.
    """
    if not os.path.exists(directory):
        os.makedirs(directory)

def cli(arguments):
    """
    Processes command line args for Snapchat FS
    @arguments Arguments dictionary obtained from docopt the library.
    """
    session = SfsSession(SETTINGS[USERNAME], SETTINGS[PASSWORD])
    session.login()
    if arguments['list']:
        list_all_downloadable_sfs_files(session)
    if arguments['download_all']:
        directory = arguments['<target_directory>']
        create_dir_if_not_exists(directory)
        download_all_sfs(session, directory)
    if arguments['download']:
        directory = arguments['<target_directory>']
        snap_ids = arguments['<snap_id>']
        create_dir_if_not_exists(directory)
        download_by_id(session, directory, snap_ids)
    if arguments['upload']:
        for filename in arguments['<file>']:
            upload_sfs_file(session, filename)

def main():
    global SETTINGS
    try:
        init_settings_from_config_file()
        arguments = docopt(__doc__, version=constants.VERSION)
        cli(arguments)
    except Exception as e:
        print util.red("Error:"),
        print e
        exit()

if __name__ == '__main__':
    main()