s2_import.py

import argparse
from scenes import utils, sentinel, save_scenes

# Arguments
parser = argparse.ArgumentParser(description="Test",)
parser.add_argument("--root_dirs", nargs='+', help="Root directories containing S2 archives (.zip)", required=True)
parser.add_argument("--out_pickle", help="Output pickle file", required=True)
parser.add_argument('--level', default='2A', const='2A', nargs='?', choices=['2A', '3A'], help='Product level')
params = parser.parse_args()

# Get all scenes in the root_dir
product_type = sentinel.Sentinel23AScene if params.level == '2A' else sentinel.Sentinel23AScene
archives = []
for root_dir in params.root_dirs:
    archives += utils.find_files_in_all_subdirs(pth=root_dir, pattern="*.zip", case_sensitive=False)
scenes = [product_type(archive=archive) for archive in archives]

# Save scenes in a pickle file
save_scenes(scenes, params.out_pickle)