database/background/get_state_set.py

import os
# print now directory
print(os.getcwd())
state_set = set()
city_set = set()
with open('database/background/citySet_with_states.txt','r') as f:
    city_set = f.read().strip().split('\n')
    for city in city_set:
        city_name = city.split('\t')[0]
        state_name = city.split('\t')[1]
        state_set.add(state_name)
        city_set.add(city_name)
        # write to new file
    f.close()
# with open('database/background/stateSet.txt', 'a') as f:
#     for state_name in state_set:
#         f.write(state_name.split('\\')[0] + '\n')
#     f.close()
with open('database/background/citySet_2.txt', 'a') as f:
    for city_name in city_set:
        f.write(city_name.split('\\')[0] + '\n')
    f.close()