IC84 - Tips for Writing Reusable Python Scripts https://www.insightcoder.com/84/5-tips-for-writing-reusable-python-scripts/

01-add_nums.py

import sys

def add_nums(nums):
    return sum(nums)

args = sys.argv[1:]
nums = [float(arg) for arg in args]
result = add_nums(nums)
print(result)

02-use_add_nums.py

import add_nums

my_nums = [1, 2, 3]
result = add_nums.add_nums(my_nums)
print(result)

03-add_nums_2.py

import sys

def add_nums(nums):
    return sum(nums)

if __name__ == '__main__':
    args = sys.argv[1:]
    nums = [float(arg) for arg in args]
    result = add_nums(nums)
    print(result)

04-delete_pyc.py

import os

def delete_pyc():
    pyc_files = []
    for (path, dirs, files) in os.walk('.'):
        for f in files:
            if f.endswith('.pyc'):
                pyc_file = os.path.join(path, f)
                os.remove(pyc_file)

if __name__ == '__main__':
    delete_pyc()

05-delete_pyc_2.py

import os
import sys

def delete_pyc(root_dir):
    pyc_files = []
    for (path, dirs, files) in os.walk(root_dir):
        for f in files:
            if f.endswith('.pyc'):
                pyc_file = os.path.join(path, f)
                os.remove(pyc_file)

if __name__ == '__main__':
    root_dir = '.'
    if len(sys.argv) > 1:
        root_dir = sys.argv[1]

    delete_pyc(root_dir)

06-cross_platform.py

import sys

class FooWindows:
    # Windows implementation
    def __init__(self):
        print('Windows')

class FooLinux:
    # Linux implementation
    def __init__(self):
        print('Linux')

Foo = FooWindows if sys.platform == 'win32' else FooLinux

07-delete_files.py

import os
import sys

root_dir = '.'
file_ext = '.pyc'

def delete_files():
    for (path, dirs, files) in os.walk(root_dir):
        for f in files:
            if f.endswith(file_ext):
                file_to_delete = os.path.join(path, f)
                os.remove(file_to_delete)

if __name__ == '__main__':
    delete_files()

08-delete_files_2.py

import os
import sys

ROOT_DIR = '.'
FILE_EXT = '.pyc'

def delete_files(root_dir, file_ext):
    for (path, dirs, files) in os.walk(root_dir):
        for f in files:
            if f.endswith(file_ext):
                file_to_delete = os.path.join(path, f)
                os.remove(file_to_delete)

if __name__ == '__main__':
    delete_files(ROOT_DIR, FILE_EXT)

09-delete_files.py

import os
import sys

ROOT_DIR = '.'
FILE_EXT = '.pyc'

def find_files(root_dir, file_ext):
    found_files = []
    for (path, dirs, files) in os.walk(root_dir):
        for f in files:
            if f.endswith(file_ext):
                found_file = os.path.join(path, f)
                found_files.append(found_file)
    return found_files

def delete_files(files_to_delete):
    for f in files_to_delete:
        os.remove(f)

if __name__ == '__main__':
    files_to_delete = find_files(ROOT_DIR, FILE_EXT)
    delete_files(files_to_delete)