generator-javagenerator.py

import json
import os
import re
import shutil

class JavaGenerator():


    re_velke_pismena = re.compile('[A-Z]')

    def typ(self,t):
        if t == "xy":
            return "java.awt.Point"
        if t == "wh":
            return "java.awt.Dimension"
        if t == "xywh":
            return "java.awt.Rectangle"
        if t == "string":
            return "String"
        if t == "variant":
            return "Object"
        if t == "keyvalue":
            return "java.util.Map"
        if t == "int32":
            return "int"
        if t == "BasicEventHandler":
            return "java.beans.EventHandler"
        if t == "inputstream":
            return "java.io.InputStream"
        if t == "NotificationEventHandler":
            return "com.sun.nio.sctp.NotificationHandler"
        if t == "ImportExportHandler":
            return "Object"
        else:
            return t


    def daj_default(self,typ):
        if typ == "boolean":
            return "false"
        if typ == "string":
            return "\"\"";
        if typ == "int32":
            return "0"
        else:
            return "null"

    def generuj_metodu(self, dok,f):
        typ = self.typ(dok['returnType'])
        name = dok['name']
        isreturn = False
        if typ:
            f.write("public "+typ+" "+name)
            isreturn = True
        else:
            f.write("public void " + name)
            isreturn =False



        if (dok['parameters']):
            count = len(dok['parameters'])
            i = 0
            f.write("(")
            for par in dok['parameters']:
                f.write(self.typ(par['type']) + " " + par['name'])
                if i != count-1:
                    f.write(", ")
                i += 1
            f.write("){\n")

        else:
            f.write("(){\n")

        if isreturn:
            f.write("\treturn "+self.daj_default(typ)+";\n")
            f.write("}\n")
        else:
            f.write("}\n")


    def generuj_property(self, dokument, f):
        name = self.konv_nazov_property(dokument['name'])
        name = str(name[0]).lower() + str(name[1:])
        if dokument['readable']:
            f.write("public ")
            f.write(self.typ(dokument['type']))
            f.write(" get")
            f.write(name+"(){\n")
            f.write("\treturn this."+name+";\n")
            f.write("}\n\n")
        if dokument['writable']:
            f.write("public ")
            f.write("void")
            f.write(" set")
            f.write(name+"("+self.typ(dokument['type']) +" "+ name+"){\n")
            f.write("\tthis."+name+" = "+name+";\n")
            f.write("}\n\n")

    def generuj_triedu(self,dokument, f):

        #dam to do nejakeho balicka
        f.write("package {0};".format("umlfri"))
        f.write("\n")

        f.write("public class "+ dokument['name']+" ")

        if dokument['base']:  # je tam nejaka hodnota?
            f.write("extends "+dokument['base'])
        f.write("{\n\n")

        #privatne atributy
        for prop in dokument['properties']:
            f.write("private "+self.typ(prop['type'])+" "+prop['name']+"; \n")

        if not dokument['properties'] and not dokument['methods']:
            f.write("\n")

        for prop in dokument['properties']:
            self.generuj_property(prop, f)


        #generovanie fcii
        for method in dokument['methods']:
            self.generuj_metodu(method, f)

        f.write("}")  # posledna zatvorka

    def konv_nazov_modulu(self, nazov_triedy):
        return nazov_triedy #nazov_triedy[0].upper() + nazov_triedy[0::].lower()

    def konv_nazov_property(self, nazov_property):
        return nazov_property[0].upper() + nazov_property[1:]


    def generuj(self, zdroj_dir, ciel_dir):

        if os.path.exists(ciel_dir):  # skontrolujem ci existuje
            shutil.rmtree(ciel_dir)  # zmazem adresar
        os.mkdir(ciel_dir)  # vytvorim
        #open(os.path.join(ciel_dir, "__init__.py"), "w").close()

        # najdem vsetky json subory
        for nazov in os.listdir(zdroj_dir):
            nazov_cesta = os.path.join(zdroj_dir, nazov)  # cela cesta
            # zistim ci je to subor
            if os.path.isfile(nazov_cesta) and nazov.endswith('.json'):
                with open(nazov_cesta) as f:
                    dokument = json.load(f)
                ciel_cesta = os.path.join(ciel_dir, self.konv_nazov_modulu(dokument['name']) + '.java')  # nazov triedy bude nazov suboru
                with open(ciel_cesta, "w") as f:
                    self.generuj_triedu(dokument, f)