Merge branch 'master' of https://github.com/nickvsnetworking/pyhss

Author: davidkneipp

lib/database.py

@@ -2168,7 +2168,7 @@ def Get_Charging_Rules(self, imsi, apn):
                 ChargingRule['apn_data'] = apn_data
 
                 #Get Charging Rules list
-                if apn_data['charging_rule_list'] == None:
+                if not apn_data['charging_rule_list'] or apn_data['charging_rule_list'].strip() == '':
                     self.logTool.log(service='Database', level='debug', message="No Charging Rule associated with this APN", redisClient=self.redisMessaging)
                     ChargingRule['charging_rules'] = None
                     return ChargingRule

tools/view_stats.py

@@ -1,4 +1,4 @@
-#This utility prints PyHSS stats stored in Redis
+# This utility prints PyHSS stats stored in Redis
 import yaml
 import sys
 with open(sys.path[0] + '/../config.yaml') as stream:
@@ -15,12 +15,13 @@
 
 print("\n\nDiameter Peers:")
 ActivePeerDict = r.get('ActivePeerDict')
-if len(ActivePeerDict) == 0:
+if ActivePeerDict is None or len(ActivePeerDict) == 0:
     print("No connected peers.")
+    sys.exit()
 ActivePeerDict = json.loads(ActivePeerDict)
 
 for keys in ActivePeerDict:
     print(keys)
     for subkeys in ActivePeerDict[keys]:
         print("\t" + str(subkeys) + ": \t" + str(ActivePeerDict[keys][subkeys]))
-    print('\n')
+    print('\n')