diadia

興味があることをやってみる。自分のメモを残しておきます。

postgres create

    
import pandas as pd
import zipfile,os,time

ZIP_DIR     = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + "\\ZIP_DIR\\"
UNZIP_DIR   = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + "\\UNZIP_DIR\\"
MERGED_CSVS = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + "\\DIR_DIR\\MERGED_CSVS\\"
zip_files = os.listdir(ZIP_DIR)


#zipファイルを解凍してrenameを実行。すべて解凍する。
userid_list = []
for zf in zipfiles:
	with zipfile.ZipFile(ZIP_DIR+zf) as zip:
		time.sleep(0.0000001)
		timestamp = str(time.time()).replace(".", "_")
		ziplst = zip.namelist()
		for f in ziplst:
			#チェック
			print(f)
			sys.exit()
			userid = f.split("_","_")[X]
			zip.extract(f, UNZIP_DIR)
			os.rename("".join([UNZIP_DIR, f]), "".join([UNZIP_DIR, userid, span, timestamp, ".csv"]))
			userid_list.append(userid)



#同一useridのcsvファイルをUNZIPディレクトリから取得
unzip_list = os.listdir(UNZIP_DIR)

for userid in set(userid_list):
	
	csv_list = [ file for file in unzip_list if userid in file ]
	csv_list.sort()
	print(csv_list)
	for f in csv_list:
		unzip_list.pop(f)

	sys.exit()

	#マージファイルを作成
	pd_list = [ pd.read_csv(csv_file) for csv_file in csv_list ]
	df = pd.concat(pd_list)
	#マージファイルタイトルの作成
	merge_file = "".join([userid, span, ".csv" ]) 
	df.to_csv("".join([MERGED_CSVS, merge_file]), index=False, encoding="utf-8")