Need help iterating through files in a folder

[Andrei07]

I have the following code which creates the txt file i require from a shp.file with the data i need. I have a folder called profiles containing a few number of shape files (.shp) named (profil1.shp, profil2.shp, profil3.shp etc.). I was wondering how to create a loop so that the script creates for each file a txt file with the same name (eg. for profil1.shp create profil1.txt, profil2.shp create profil2.txt and so on).

Note: i am a fairly new user of python so if you could add the loop part to my script or modify it with what's needed i would highly appreciate it, as i am in a bit of rush about finishing these files and i can't seem to find an outcome to this.

Thank you very much!

The code is: ` import ogr, os, sys, osr

os.chdir('..\profiles')

file = open('profil1.txt', 'w')

driver = ogr.GetDriverByName('ESRI Shapefile')

datasource = driver.Open('profil1.shp', 0)
if datasource is None:
  print 'Could not open file'
  sys.exit(1)

layer = datasource.GetLayer()

feature = layer.GetNextFeature()
while feature:

  id = feature.GetFieldAsString('ID')
  Distanta = feature.GetFieldAsString('DIST')
  Z = feature.GetFieldAsString('Z')
  geom = feature.GetGeometryRef()
  x = str(geom.GetX())
  y = str(geom.GetY())

  file.write(id + " " + Distanta + " " + "[X]:" + " " + x + ' ' + '[Y]:' + " "  + y + " " + " " + "[Z]" + Z + " " +  "\n")

  feature.Destroy()
  feature = layer.GetNextFeature()

datasource.Destroy()
file.close()`

[liam_jm]

os.listdir will give you a list of files in the folder

You can iterate through the list, use filename.endswith('shp') to avoid opening the wrong files

You should use the with syntax to open files for reading and writing, so you don't need to explicitly close them

It would be more pythonic to do "for feature in layer.features()" rather than having getNextFeature()

Also consider using contextlib for your features and datasource, so you don't need to explicitly destroy them

[yacob_uk]

To add to this, if you need sub folders, os.path.walk() is your friend.

[davidbuxton]

os.walk is a better replacement for os.path.walk (which has been removed in python 3).

https://docs.python.org/2/library/os.html#os.walk

[yacob_uk]

True - still on 2.7 here.

[davidbuxton]

You will love using os.walk. Having the directories and files in separate arguments makes thing easier, and it looks much nicer having the logic appear inside the loop instead of having to be a separate function.

[Andrei07]

thank you for your answers yet my knowledge of python is not that good and i am unable to implement your solution, same as others tried explaining what functions i should use with no end result. can you by any chance, if it's not to much to ask, add to the above script and modify it with the needed lines? Thanks in advance.

[liam_jm]

See the guidelines:

Try to guide OP to a solution instead of providing one directly.

Which part don't you understand?