Forráskód Böngészése

str.startswith instead of contains

vnc-console 7 hónapja
szülő
commit
2f02f05f8a
1 módosított fájl, 5 hozzáadás és 1 törlés
  1. 5 1
      pymt940/CSV2MentorBtrl.py

+ 5 - 1
pymt940/CSV2MentorBtrl.py

@@ -136,9 +136,13 @@ class PaymentsProcessorBtrl:
         df['Transaction details'] = df[lang['Description']]
         df['day'] = df[lang['Transaction date']].str.slice(*day_slice)
         # df['symbols'] = ''
-        df.loc[df['Transaction details'].str.lower().str.contains('comision'), 'symbols'] = 627
+        df.loc[df['Transaction details'].str.lower().str.startswith('comision'), 'symbols'] = 627
+        df.loc[df['Transaction details'].str.lower().str.startswith('abonament'), 'symbols'] = 627
         df.loc[df['symbols'] == 627, 'types'] = 'DIRECT PE CHELTUIELI'
 
+        df.loc[df['Transaction details'].str.lower().str.startswith('dobanda'), 'symbols'] = 766
+        df.loc[df['symbols'] == 766, 'types'] = 'DIRECT PE VENITURI'
+
         unique_count = df['day'].nunique()
         self.doc_list = list(range(1, unique_count + 1))
         days = df['day'].unique().tolist()