add_license_pub.py

#!/usr/bin/python3
#coding:utf-8

# Copyright (c) 2021-2022 Amlogic, Inc. All rights reserved.

# SPDX-License-Identifier: MIT

import os
import sys
import string
import re

def add_Header_01(filepath, filename):
    if os.path.exists(filepath) :
        head_info = string.Template(
'''/*
 * Copyright (c) 2021-2022 Amlogic, Inc. All rights reserved.
 *
 * SPDX-License-Identifier: MIT
 */\n
''')
        head = head_info.substitute(vars())
        f = open(filepath, "r+", errors='ignore')
        content = f.read()

        f.seek(0,0)
        new_content = re.sub("\r\n", "\n", content)
        f.write(head)
        f.write(new_content)
        f.close


def add_Header_02(filepath, filename):
    if os.path.exists(filepath) :
        head_info = string.Template(
'''# Copyright (c) 2021-2022 Amlogic, Inc. All rights reserved.

# SPDX-License-Identifier: MIT\n
''')
        head = head_info.substitute(vars())
        f = open(filepath, "r+", errors='ignore')
        content = f.read()

        f.seek(0,0)
        new_content = re.sub("\r\n", "\n", content)
        f.write(head)
        f.write(new_content)
        f.close

def del_C_Header(filepath):
    if os.path.exists(filepath) :
        file = open(filepath, "r", errors='ignore')
        lines = file.readlines()
        beforeTag = True
        writer = open(filepath, 'w', errors='ignore')
        for line in lines :
            if '#include' in line:
                beforeTag = False
            if beforeTag == False:
                writer.write(line)

def del_H_Header(filepath):
    if os.path.exists(filepath):
        file = open(filepath, "r", errors='ignore')
        lines = file.readlines()
        beforeTag = True
        writer = open(filepath, 'w', errors='ignore')
        for line in lines :
            if '#ifndef ' in line or '#include ' in line or '#define ' in line:
                beforeTag = False
            if beforeTag == False:
                writer.write(line)

if __name__ == '__main__':
    path=sys.argv[1]
    list = os.walk(path, True)
    for dir in list:
        files = dir[2]
        for file in files:
            filepath = os.path.join(dir[0], file)
            print(filepath)
            if 'Kconfig' in file or 'CMakeList' in file:
                add_Header_02(filepath, file)
            elif '.cmake' in file:
                add_Header_02(filepath, file)
            elif '.h' in file or '.ld' in file:
                del_H_Header(filepath)
                add_Header_01(filepath, file)
            elif '.c' in file:
                del_C_Header(filepath)
                add_Header_01(filepath, file)

    print( 'Complete!!!!!!!!!!!!!!!')