Refactor default country prefix (+47) into a global variable

jherland · Mar 12, 2011 · ffac3b4 · ffac3b4
1 parent 245e315
commit ffac3b4
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sms_query.py b/sms_query.py
@@ -86,6 +86,8 @@
 
 DbFilename = "sms.db" # On Nokia N900: /home/user/.rtcom-eventlogger/el-v1.db
 
+CountryPrefix = "+47" # Default phone number country prefix
+
 AnsiColors = {
 	"red":     "\033[91m",
 	"green":   "\033[92m",
@@ -223,10 +225,10 @@ def args (self):
 
 	def add (self, phonenum):
 		self.nums.append(phonenum)
-		if phonenum.startswith("+47"):
-			self.nums.append(phonenum[3:])
+		if phonenum.startswith(CountryPrefix):
+			self.nums.append(phonenum[len(CountryPrefix):])
 		else:
-			self.nums.append("+47" + phonenum)
+			self.nums.append(CountryPrefix + phonenum)
 
 
 def main (args = []):