from os import system

def get_building_names(filename):
  result = []
  csvfile = open(filename, 'r')
  for line in csvfile:
    linelist = line.strip('\n').split('.')
    buildingname = linelist[0]
    result.append(buildingname)
  return result


#filenames = ['import_ba0.csv', 'import_ba1.csv', 'import_ba2.csv', 'import_ba3.csv', 'import_ba4.csv', 'import_salchau.csv']
filenames = ['import_fehlerhafte_geb_20221005.csv']

buildingnames = []

for filename in filenames:
  buildingnames.extend(get_building_names(filename))

query = 'select id, name from ism.ism_main_structure_element where'

first = True

for buildingname in buildingnames:
  if first == True:
    query = query + ' name = \'' + buildingname + '\''
    first = False
  else:
    query = query + ' or name = \'' + buildingname + '\''

query = query

psqlcmd = 'psql -h EXCDBS01 -d lts_ets_pro -U postgres -c "\copy (' + query + ') to \'/opt/BuildingModels/StaticMeshes/ids2\' with csv delimiter \',\';"'

system(psqlcmd)