script_count.py

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