import psycopg

db_string_test = 'TEST_DATABASE'

with psycopg.connect(user="root", password="rootroot",
                    host='localhost',
                    port='5432',
                    dbname=db_string_test,
                    sslmode='disable'
                    ) as connection:
    
    connection.autocommit = True

    with connection.cursor() as cursor:
        cursor.execute("create vertex type `IMAGE` if not exists;")
        for i in range(10):
            query = 'INSERT INTO `IMAGE` RETURN @rid;'
            cursor.execute(query)
            cursor.fetchall()
            
        print('ok for sql')
        
        for i in range(10):
            query = '{cypher}CREATE  (c:IMAGE) RETURN id(c)'
            cursor.execute(query)
            cursor.fetchall()
        print('ok for cypher')
        
        for i in range(10):
            query = '{sqlscript}INSERT INTO `IMAGE` RETURN @rid;'
            cursor.execute(query)
            cursor.fetchall()

        print('ok for sqlscript')