Merges auto generated code from multiple files into a single swift class

swift_package_merger.py

import os

__author__ = 'carlozamagni'


BASE_PATH = os.getcwd()


def merge_files(path, package_name):
    files = [f for f in os.listdir(path) if os.path.isfile(os.path.join(path,f))]

    package_code = ''

    header = 'public struct {0}'.format(package_name) + '{\n\r'

    for f in files:
        with open(os.path.join(path,f)) as current:
            content = ''.join(line for line in current)

            package_code += content

    package_code = '{}{}{}'.format(header, package_code, '}')

    print(package_code)

    with open(os.path.join(BASE_PATH, '{}.swift'.format(package_name)), 'wb') as outfile:
        outfile.write(package_code)


if __name__ == '__main__':

    for package in os.listdir(BASE_PATH):

        if '.idea' in package:
            continue

        path = os.path.join(BASE_PATH, package)

        if os.path.isdir(path) is True:
            merge_files(path, package)