Initial Project Commit
[retroconv] / csv_service.py
diff --git a/csv_service.py b/csv_service.py
new file mode 100644 (file)
index 0000000..b50f24d
--- /dev/null
@@ -0,0 +1,61 @@
+import csv
+from recv import recv
+from sent import sent
+from contactsServ import *
+
+class csvServ:
+       def __init__(self, recvFile, sentFile):
+               ifile = open(recvFile, "r")
+               reader=csv.reader(ifile)
+               self.recvLst=[]
+               self.recvNumbers=[]
+               for row in reader:
+                       self.recvLst.append(recv(row[2],row[5], row[7]))
+                       self.recvNumbers.append(row[2][-7:])
+               
+               #print "WILL"
+               
+               #self.recvLst = z.populateNames(self.recvLst)
+               
+               
+               ifile.close()
+               
+               ##############################################
+               
+               ifile = open(sentFile, "r")
+               reader=csv.reader(ifile)
+               self.sentLst=[]
+               self.sentNumbers=[]
+               for row in reader:
+                       self.sentLst.append(sent(row[3],row[5], row[7]))
+                       self.sentNumbers.append(row[3][-7:])
+               
+               
+               
+               abook=address_book()
+               
+               abook.populateNames(self.recvLst, self.sentLst)
+               
+               
+               
+               
+       def getAllReceivedMsgs(self):
+               msgs=[]
+               for msg in self.recvLst:
+                       msgs.append(msg.getSummary())
+               return msgs
+               
+       
+       
+       
+       def getMsg(self, msgNum, msgType = 0):
+               #msgType 0: recv, 1:sent
+               
+               if msgType == 0:
+                       return self.recvLst[msgNum]
+               
+               return self.sentLst[msgNum]
+               
+       
+       def getMsgText(self, msgNum):
+               return self.recvLst[msgNum].getMsg()