anumaurya114exp commited on
Commit
a39e129
·
1 Parent(s): 8a4e4ca

added get csv file and remove temporary csv files

Browse files
Files changed (1) hide show
  1. persistStorage.py +17 -1
persistStorage.py CHANGED
@@ -3,9 +3,10 @@ import sqlite3
3
  from datetime import datetime, timedelta
4
  import pytz
5
  import os
6
- from config import HUGGING_FACE_TOKEN, TABLES_DATA_DIR, logsDir
7
  import pandas as pd
8
  import csv
 
9
 
10
  try:
11
  os.makedirs(TABLES_DATA_DIR, exist_ok=True)
@@ -17,6 +18,21 @@ TIME_ZONE = 'US/Pacific'
17
  TIMEZONE_OBJ = pytz.timezone(TIME_ZONE)
18
  CACHE_TIME_EXPIRE = 20 #days
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  def append_dict_to_csv(file_path, row_data):
21
  fieldnames = row_data.keys()
22
 
 
3
  from datetime import datetime, timedelta
4
  import pytz
5
  import os
6
+ from config import HUGGING_FACE_TOKEN, TABLES_DATA_DIR, logsDir, RESULT_CSV_DIR
7
  import pandas as pd
8
  import csv
9
+ import random
10
 
11
  try:
12
  os.makedirs(TABLES_DATA_DIR, exist_ok=True)
 
18
  TIMEZONE_OBJ = pytz.timezone(TIME_ZONE)
19
  CACHE_TIME_EXPIRE = 20 #days
20
 
21
+ def getNewCsvFilePath():
22
+ fileName = "ResultCsv_" + "".join([str(random.randint(0,9)) for i in range(3)])
23
+ fileNameWithpath = os.path.join(RESULT_CSV_DIR, fileName)
24
+ return fileNameWithpath
25
+
26
+ def removeAllCsvFiles():
27
+ files = os.listdir(RESULT_CSV_DIR)
28
+ for fileName in files:
29
+ fileNameWithPath = os.path.join(RESULT_CSV_DIR, fileName)
30
+ try:
31
+ os.remove(fileNameWithPath)
32
+ except:
33
+ pass
34
+
35
+
36
  def append_dict_to_csv(file_path, row_data):
37
  fieldnames = row_data.keys()
38