scsync.py

#!/usr/bin/env python3
# coding: utf-8
"""
    scsync.module
    ~~~~~~~~~~~~~

    A secure cloud sync application.

    :copyright: 2018 Andreas Ebner, Julien Schmidt, Paul Schmiedmayer
    :license: TBD
"""

import argparse
import logging
import os
import sys

import client
import server


# Client: scsync [-h <hostname|ip-addr>] [-p <port>] [-f <directory-path>]
# Server: scsync [-s] [-p <port>]

def main():
    """
    Initializes the program by parsing the command-line flags and either running
    the client or the server module.
    """
    if sys.platform == 'win32':
        print("Windows is currently not supported.")
        sys.exit(1)

    # parse flags
    parser = argparse.ArgumentParser(
        description='Cloud Sync', conflict_handler='resolve')

    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('-h', dest='host', action='store',
                       default='localhost', help='remote host (hostname or IP address)')
    group.add_argument('-s', dest='server', action='store_true',
                       default=False, help='run in server mode')

    parser.add_argument('-p', dest='port', action='store',
                        default=5000, type=int, help='port number')
    parser.add_argument('-f', dest='path', action='store',
                        default=os.getcwd(), help='directory path containing files')
    parser.add_argument('-cc', dest='cc', action='store',
                        default=2000, type=int, help='packets per second')

    parser.add_argument('-u', dest='user', action='store',
                        default='', help='client username')
    parser.add_argument('-pass', dest='password', action='store',
                        default='', help='client user password')

    # Should be adjusted to MTU later
    parser.add_argument('-cs', dest='chunk_size', action='store',
                        default=8192, type=int, help='chunk size when transfering data')

    group = parser.add_mutually_exclusive_group(required=False)
    group.add_argument('--verbose', dest='verbose', action='store_true',
                       default=False, help='verbose output')
    group.add_argument('--debug', dest='debug', action='store_true',
                       default=False, help='verbose debug output')

    args = parser.parse_args()
    args.test = False

    # normalize path
    args.path = os.path.abspath(args.path) + '/'

    # Logging Modes
    level = logging.WARNING
    if args.verbose:
        level = logging.INFO
    elif args.debug:
        level = logging.DEBUG
    logging.basicConfig(format='[%(levelname)s] %(message)s', level=level)

    # Start Program
    if args.server:
        server.run(args)
    else:
        if args.user == '':
            print('User and password must be set!')
            sys.exit(1)
        client.run(args)

if __name__ == "__main__":
    main()