Exemplo n.º 1
0
            className = dic['className: ']
            param = dic['param: ']
            javadocComment = dic['javadocComment: ']
            blockComment = dic['blockComment: ']

            update_by_className_methodName_param(className, methodName, param,
                                                 code, javadocComment,
                                                 blockComment)
            counter += 1
            if counter > step:
                counter = 0
                session.commit()
        session.commit()


if __name__ == "__main__":
    schema_name = 'domainkg'
    engine = EngineFactory.create_engine_by_schema_name(schema_name)
    session = EngineFactory.create_session(engine=engine, autocommit=False)
    read_json_by_line('traces.json')

    # trace_list = session.query(Traces).filter_by()
    # for trace in trace_list:
    #     try:
    #         className = trace.className
    #         methodName = trace.methodName
    #         param = trace.param
    #         trace.code = 'code'
    #     except Exception:
    #         print("error")
Exemplo n.º 2
0
Arquivo: model.py Projeto: PaiXue/temp
            try:
                return session.query(POIMethod).filter_by(
                    package_name=self.package_name,
                    class_name=self.class_name,
                    return_type=self.return_type,
                    method_name=self.method_name,
                    description=self.description,
                    type=self.type).first()
            except Exception:
                traceback.print_exc()
                session.rollback()
            return None

    def find_or_create(self, session, autocommit=True):
        remote_instance = self.get_remote_object(session)
        if not remote_instance:
            session.add(self)
            if autocommit:
                session.commit()
            return self
        else:
            return remote_instance


if __name__ == "__main__":
    engine = EngineFactory.create_engine_by_schema_name('domainkg')
    metadata = MetaData(bind=engine)

    # create the table
    Base.metadata.create_all(bind=engine)