Programiranje

Python, pomoć sa skriptom

Zvonzi uto 27.7.2021 19:56

pozdrav ekipica.

 

radim na jednom hobi projektiću (react app), i naletio sam u bespućima interneta na jednu python skriptu, a znanje pythona mi je nepostojeće, koja bi mi full olakšala "posao" i skratila muke.

 

skripta iz arraya csv fajlova stvara jedan JSON file. problem je u tome što ja imam stotine csv fajlova koje želim mergati u jedan JSON. ono što me zanima jest na koji način napuniti taj array sa tih nekoliko stotina filenameova, a da nije ručni c/p? svi fileovi su u istom folderu, imaju istu ekstenziju.

 

kod skripte:

 

import os
import csv
import json

fname=["2015.csv", "2016.csv", "2017.csv", "2018.csv", "2019.csv"]

data=[]
json_str_start=""
json_str=""
for f in fname:
data_path = os.path.join("resources", f)
data_path2 = os.path.join("resources", "file.json")
with open(data_path, 'r') as csvfile:
reader = csv.reader(csvfile, delimiter=',')
year=f.replace('.csv', '')
json_str_start='{"' + year +'": '

data_list=list()

for row in reader:
data_list.append(row)

data = [dict(zip(data_list[0],row)) for row in data_list]
data.pop(0)

json_str = json_str + json_str_start + json.dumps(data) + " }, \n"

 


print (json_str[0:50])

with open(data_path2, 'w+') as jsonfile:

jsonfile.write(json_str)