from __future__ import print_function

import os  # Load the library module

# Script Name		: script_count.py
# Author				: Craig Richards
# Created				: 27th February 2012
# Last Modified		: 20th July 2012
# Version				: 1.3
# Modifications		: 1.1 - 28-02-2012 - CR - Changed inside github and development functions, so instead of if os.name = "posix" do this else do this etc
#							: I used os.path.join, so it condensed 4 lines down to 1
#							: 1.2 - 10-05-2012 - CR - Added a line to include PHP scripts.
#							: 1.3 - 20-07-2012 - CR - Added the line to include Batch scripts
# Description			: This scans my scripts directory and gives a count of the different types of scripts

path = os.getenv("scripts")  # Set the variable path by getting the value from the OS environment variable scripts
dropbox = os.getenv("dropbox")  # Set the variable dropbox by getting the value from the OS environment variable dropbox


def clear_screen():  # Function to clear the screen
    if os.name == "posix":  # Unix/Linux/MacOS/BSD/etc
        os.system('clear')  # Clear the Screen
    elif os.name in ("nt", "dos", "ce"):  # DOS/Windows
        os.system('CLS')  # Clear the Screen


def count_files(path,
                extensions):  # Start of the function to count the files in the scripts directory, it counts the extension when passed below
    counter = 0  # Set the counter to 0
    for root, dirs, files in os.walk(path):  # Loop through all the directories in the given path
        for file in files:  # For all the files
            counter += file.endswith(extensions)  # Count the files
    return counter  # Return the count


def github():  # Start of the function just to count the files in the github directory
    github_dir = os.path.join(dropbox, 'github')  # Joins the paths to get the github directory - 1.1
    github_count = sum((len(f) for _, _, f in os.walk(github_dir)))  # Get a count for all the files in the directory
    if github_count > 5:  # If the number of files is greater then 5, then print the following messages

        print('\nYou have too many in here, start uploading !!!!!')
        print('You have: ' + str(github_count) + ' waiting to be uploaded to github!!')
    elif github_count == 0:  # Unless the count is 0, then print the following messages
        print('\nGithub directory is all Clear')
    else:  # If it is any other number then print the following message, showing the number outstanding.
        print('\nYou have: ' + str(github_count) + ' waiting to be uploaded to github!!')


def development():  # Start of the function just to count the files in the development directory
    dev_dir = os.path.join(path, 'development')  # Joins the paths to get the development directory - 1.1
    dev_count = sum((len(f) for _, _, f in os.walk(dev_dir)))  # Get a count for all the files in the directory
    if dev_count > 10:  # If the number of files is greater then 10, then print the following messages

        print('\nYou have too many in here, finish them or delete them !!!!!')
        print('You have: ' + str(dev_count) + ' waiting to be finished!!')
    elif dev_count == 0:  # Unless the count is 0, then print the following messages
        print('\nDevelopment directory is all clear')
    else:
        print('\nYou have: ' + str(
            dev_count) + ' waiting to be finished!!')  # If it is any other number then print the following message, showing the number outstanding.


clear_screen()  # Call the function to clear the screen

print('\nYou have the following :\n')
print('AutoIT:\t' + str(
    count_files(path, '.au3')))  # Run the count_files function to count the files with the extension we pass
print('Batch:\t' + str(count_files(path, ('.bat', ',cmd'))))  # 1.3
print('Perl:\t' + str(count_files(path, '.pl')))
print('PHP:\t' + str(count_files(path, '.php')))  # 1.2
print('Python:\t' + str(count_files(path, '.py')))
print('Shell:\t' + str(count_files(path, ('.ksh', '.sh', '.bash'))))
print('SQL:\t' + str(count_files(path, '.sql')))

github()  # Call the github function
development()  # Call the development function