def importEntities(conn, cur, sg): debug.debug("starting import Entities", debug.INFO) entities = sg.schema_entity_read() classes = entities.keys() classes.sort() for entityType in classes: if entityType in ["EventLogEntry"]: continue if len(UPDATE_ONLY) > 0 and entityType not in UPDATE_ONLY: continue entityName = cleanSysName(entities[entityType]["name"]["value"]) if entityType.endswith("Connection"): entityName = entityType debug.debug("import entities of type " + entityType) fieldList = connectors.getClassOfType(entityName).shotgun_fields debug.debug("deleting entities of type " + entityType) query = "DELETE FROM \"%s\"" % (entityType) cur.execute(query) debug.debug("loading entities of type " + entityType) objects = sg.find(entityType, [["id", "greater_than", 0]], fieldList.keys()) for obj in objects: values = [] names = [] reprs = [] for fieldName in fieldList.keys(): sgType = fieldList[fieldName]['data_type']['value'] convFunc = connectors.getConversionSg2Pg(sgType) if convFunc != None: names.append("\"%s\"" % fieldName) if sgType == "image" and obj[fieldName] != None: thumbnails.saveShotgunImageLocally(obj[fieldName]) if sgType == "multi_entity": reprs.append("%s::entity_sync[]") else: reprs.append("%s") values.append(convFunc(obj[fieldName])) query = "INSERT INTO \"%s\" (%s) VALUES (%s)" % ( entityType, ", ".join(names), ", ".join(reprs)) debug.debug(cur.mogrify(str(query), values), debug.DEBUG) cur.execute(query, values) conn.commit() debug.debug("finnished import Entities", debug.INFO)
def importEntities(conn, cur, sg): debug.debug("starting import Entities", debug.INFO) entities = sg.schema_entity_read() classes = entities.keys() classes.sort() for entityType in classes: if entityType in ["EventLogEntry"]: continue if len(UPDATE_ONLY) > 0 and entityType not in UPDATE_ONLY: continue entityName = cleanSysName(entities[entityType]["name"]["value"]) if entityType.endswith("Connection"): entityName = entityType debug.debug("import entities of type " + entityType) fieldList = connectors.getClassOfType(entityName).shotgun_fields debug.debug("deleting entities of type " + entityType) query = 'DELETE FROM "%s"' % (entityType) cur.execute(query) debug.debug("loading entities of type " + entityType) objects = sg.find(entityType, [["id", "greater_than", 0]], fieldList.keys()) for obj in objects: values = [] names = [] reprs = [] for fieldName in fieldList.keys(): sgType = fieldList[fieldName]["data_type"]["value"] convFunc = connectors.getConversionSg2Pg(sgType) if convFunc != None: names.append('"%s"' % fieldName) if sgType == "image" and obj[fieldName] != None: thumbnails.saveShotgunImageLocally(obj[fieldName]) if sgType == "multi_entity": reprs.append("%s::entity_sync[]") else: reprs.append("%s") values.append(convFunc(obj[fieldName])) query = 'INSERT INTO "%s" (%s) VALUES (%s)' % (entityType, ", ".join(names), ", ".join(reprs)) debug.debug(cur.mogrify(str(query), values), debug.DEBUG) cur.execute(query, values) conn.commit() debug.debug("finnished import Entities", debug.INFO)
def test_retrieval_simple( self ): localpath = thumbnails.saveShotgunImageLocally( self.testnode.image ) results = os.stat( localpath ) debug.debug(results.st_mtime) self.assertTrue(results.st_mtime > time.time() - 20, "file does not seem to have been written lately")