source/blender/makesrna/rna_cleanup/rna_cleaner_merge.py

   1 #!/usr/bin/env python3
   2
   3 import sys
   4
   5 '''
   6 Example usage:
   7  python3 rna_cleaner_merge.py out_work.py rna_booleans_work.py
   8 '''
   9
  10
  11 def main():
  12
  13     def work_line_id(line):
  14         return line[2].split("|")[-1], line[3]  # class/from
  15
  16     if not (sys.argv[-1].endswith(".py") and sys.argv[-2].endswith(".py")):
  17         print("Only accepts 2 py files as arguments.")
  18
  19     sys.path.insert(0, ".")
  20
  21     mod_from = __import__(sys.argv[-1][:-3])
  22     mod_to = __import__(sys.argv[-2][:-3])
  23
  24     mod_to_dict = dict([(work_line_id(line), line) for line in mod_to.rna_api])
  25     mod_from_dict = dict([(work_line_id(line), line) for line in mod_from.rna_api])
  26
  27     rna_api_new = []
  28
  29     for key, val_orig in mod_to_dict.items():
  30         try:
  31             val_new = mod_from_dict.pop(key)
  32         except:
  33             # print("not found", key)
  34             val_new = val_orig
  35
  36         # always take the class from the base
  37         val = list(val_orig)
  38         val[0] = val_new[0]  # comment
  39         val[4] = val_new[4]  # -> to
  40         val = tuple(val)
  41         rna_api_new.append(val)
  42
  43     def write_work_file(file_path, rna_api):
  44         rna_api = list(rna_api)
  45         rna_api.sort(key=work_line_id)
  46         file_out = open(file_path, "w")
  47         file_out.write("rna_api = [\n")
  48         for line in rna_api:
  49             file_out.write("    %s,\n" % (repr(line)))
  50         file_out.write("]\n")
  51         file_out.close()
  52
  53     file_path = sys.argv[-2][:-3] + "_merged.py"
  54     write_work_file(file_path, rna_api_new)
  55
  56     if mod_from_dict:
  57         file_path = sys.argv[-2][:-3] + "_lost.py"
  58         write_work_file(file_path, list(mod_from_dict.values()))
  59         print(
  60             "Warning '%s' contains lost %d items from module %s.py" %
  61             (file_path, len(mod_from_dict), mod_from.__name__)
  62         )
  63
  64
  65 if __name__ == "__main__":
  66     main()