Python script to create a zip of a folder, without .DS_Store files. Suggestions/comments welcome!

zip.py

"""
MIT License.
"""

import os
import sys
import zipfile

def zip_directory(dir_to_zip):
    # Change to the parent directory of the directory to zip
    os.chdir(os.path.dirname(dir_to_zip))

    # Get the base name of the directory to zip
    target = os.path.basename(dir_to_zip)

    print(target)

    # Remove the existing zip file if it exists
    zip_filename = f"{target}.zip"
    if os.path.exists(zip_filename):
        os.remove(zip_filename)

    # Create a new zip file
    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(target):
            for file in files:
                if file != '.DS_Store':
                    zipf.write(os.path.join(root, file),
                               os.path.relpath(os.path.join(root, file), target))

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python zip.py <directory_to_zip>")
        sys.exit(1)

    dir_to_zip = sys.argv[1]
    zip_directory(dir_to_zip)