- Fix: replace 'TOTTRDVAL' with 'TOTTRDQTY'. Resolves issue #62

BennyThadikaran · Nov 1, 2023 · d22e1ba · d22e1ba
1 parent 8bf8508
commit d22e1ba
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/defs/defs.py b/src/defs/defs.py
@@ -255,16 +255,19 @@ def toAmiBrokerFormat(file: Path):
     'Converts and saves bhavcopy into amibroker format'
 
     cols = ['SYMBOL', 'TIMESTAMP', 'OPEN', 'HIGH',
-            'LOW', 'CLOSE', 'TOTTRDVAL', 'ISIN']
+            'LOW', 'CLOSE', 'TOTTRDQTY', 'ISIN']
 
     rcols = list(cols)
     rcols[1] = 'DATE'
     rcols[-2] = 'VOLUME'
 
     df = read_csv(file, parse_dates=['TIMESTAMP'])
 
-    df = df[(df['SERIES'] == 'EQ') | (
-        df['SERIES'] == 'BE') | (df['SERIES'] == 'BZ')]
+    df = df[(df['SERIES'] == 'EQ') |
+            (df['SERIES'] == 'BE') |
+            (df['SERIES'] == 'BZ') |
+            (df['SERIES'] == 'SM') |
+            (df['SERIES'] == 'ST')]
 
     df = df[cols]
     df.columns = rcols