update_wxstocks.py

import sys, os, wx, stat, hashlib, shutil, platform, logging

OS_TYPE = platform.system() # Detect OS, for file directory purposes
OS_TYPE_INDEX = None
try:
    supported_os_types = ["Darwin", "Linux", "Windows"]
    OS_TYPE_INDEX = supported_os_types.index(OS_TYPE)
except Exception as e:
    logging.warning(e)
    logging.warning("This program has not be properly formatted for your OS. Edit the config file if needed.")
    sys.exit()

if OS_TYPE_INDEX == 2: #Windows
    slash = "\\"
else:
    slash = "/"

current_dir = os.path.dirname(os.path.realpath(__file__))

app = wx.App()
info_label = "Update wxStocks"
info_text = "You are about to update wxStocks.\nFirst, this update file should be run from your existing wxStocks folder.\nNext, you will select the folder of the new version of wxStocks.\nNote that the new version's folder should not be zipped.\nRegardless, none of your data files should be changed, nor should any of your user files be changed."
info = wx.MessageBox(info_text, info_label, wx.OK | wx.ICON_INFORMATION)
dialog = wx.DirDialog(None, "Choose a directory:",style=wx.DD_DEFAULT_STYLE | wx.DD_NEW_DIR_BUTTON)
if dialog.ShowModal() == wx.ID_OK:
    newly_downloaded_path = dialog.GetPath()
else:
    sys.exit()
dialog.Destroy()

# check to see if selected folder is a wxStocks folder
if not (os.path.isfile(newly_downloaded_path + slash + "wxStocks.py") and os.path.isfile(newly_downloaded_path + slash + "update_wxstocks.py")):
    # if wxStocks.py or update_wxstocks.py don't exist
    logging.warning("It appears you have not selected a wxStocks folder.")
    logging.warning("Please try again, selecting an unzipped version of wxStocks.")
    logging.warning("In the case of a change to the updating process, please check the wxStocks readme file in your new folder.")
    sys.exit()

# get new directory split name
new_split_folder_name = (newly_downloaded_path + slash + "wxStocks.py").split(slash)[-2]

# get current directory split name
for root, dirs, files in os.walk(current_dir):
    for filename in files:
        if filename.endswith("wxStocks.py"):
            current_path = os.path.join(root, filename)
            path_list = current_path.split(slash)
            old_split_folder_name = path_list[-2]
            break

# add new files
user_directories = ["DO_NOT_COPY", "user_data", "wxStocks_data", "functions_for_custom_analysis_go_in_here"]
for root, dirs, files in os.walk(newly_downloaded_path):
    for filename in files:
        if any(folder_name in root for folder_name in user_directories):
            continue
        if filename.endswith(".py"):
            #logging.warning(root)
            newly_downloaded_file_path = os.path.join(root, filename)

            split_root_path =  root.split(slash + new_split_folder_name + slash)
            try:
                full_extention = split_root_path[1]
            except:
                full_extention = None

            #logging.warning(split_root_path)
            if full_extention:
                existing_file_path = slash.join([current_dir, full_extention]) + slash + filename
            else:
                existing_file_path = slash.join([current_dir, filename])
            #logging.warning(existing_file_path)

            try:
                checksum_new = hashlib.md5(open(str(newly_downloaded_file_path)).read()).hexdigest()
            except:
                checksum_new = None
            try:
                checksum_old = hashlib.md5(open(str(existing_file_path)).read()).hexdigest()
            except:
                checksum_old = None

            if checksum_new:
                if checksum_new != checksum_old:
                    logging.warning("UPDATING:", existing_file_path)
                    shutil.copy(newly_downloaded_file_path, existing_file_path)
                    # remove existing .pyc file
                    pyc_path = os.path.splitext(existing_file_path)[0] + ".pyc"
                    try:
                        os.remove(pyc_path)
                    except:
                        pass
                    logging.warning('"' + filename + '"', "was successfully updated")

# delete nonexistant file from current directory
for root, dirs, files in os.walk(current_dir):
    for filename in files:
        if root.split(slash)[-1] in ["DO_NOT_COPY", "user_data", "wxStocks_data"]:
            continue
        if filename.endswith(".py"):
            #logging.warning(root)
            existing_file_path = os.path.join(root, filename)

            split_root_path =  root.split(slash + old_split_folder_name + slash)
            try:
                full_extention = split_root_path[1]
            except:
                full_extention = None

            #logging.warning(split_root_path)
            if full_extention:
                potential_new_file_path = slash.join([newly_downloaded_path, full_extention]) + slash + filename
            else:
                potential_new_file_path = slash.join([newly_downloaded_path, filename])
            if not os.path.isfile(potential_new_file_path): # file no longer exists
                logging.warning("Deleting:", existing_file_path)
                logging.warning("It apparently no longer exists")
                try:
                    os.remove(existing_file_path)
                    logging.warning('"' + filename + '"', "was successfully removed")
                except:
                    pass
                # remove existing .pyc file
                pyc_path = os.path.splitext(existing_file_path)[0] + ".pyc"
                try:
                    os.remove(pyc_path)
                except:
                    pass
                # delete directory if it is empty
                if not os.listdir(root):
                    logging.warning("Removing empty directory:", root)
                    os.rmdir(root)

logging.warning("Update complete")