Beispiel #1
0
	def run(self,input,sender):
		regex = '(.*)\s%s\s+(\S*)' % self.keyword
		method_checker = re.match(regex,input)
		response = 'default'
		if(method_checker.group(2) == 'update'):
			""" banana: basecamp update"""
			self.cacheData()
			response = 'Updated!'

		elif(method_checker.group(2) == 'log'):
			""" banana:: basecamp log <email> <"project name"> <hours> <"desc">"""
			regex = '(.*)\s%s\s+log\s+(.*)\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.+[0-9])\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)
			if parser:
				projectid = ''
				projectname = parser.group(3)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
				
				projectid = int(project['id'])

				hour = parser.group(4)
				desc = parser.group(5)
				date = int(time.strftime("%Y%m%d"))

				user_email = parser.group(2).split('|')[1][:-1]
				userid = ''

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				xml = bc.people().text
				items = ET.fromstring(xml).findall('person')
				for item in items:
					if item.find('email-address').text == user_email:
						userid = item.find('id').text
						break
				
				if userid == '':
					response = 'Email not recognized!'
					return response

				projectpeople = bc.people_per_project(projectid).text
				peoplesearch = ET.fromstring(projectpeople).findall('person')
				for person in peoplesearch:
					if int(person.find('id').text) == userid:
						bc.create_time_entry(desc,float(hour),int(userid),date,int(projectid),None).text
						response = 'Logged successfully!'
						return response

				response = 'You do not belong to that project.'
				return response


			""" banana:: basecamp log <"project name"> <hours> <"desc">"""
			regex = '(.*)\s%s\s+log\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.+[0-9])\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)
			
			if parser:
				print "log 2"
				projectid = ''
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
				
				projectid = project['id']

				hour = parser.group(3)
				desc = parser.group(4)
				date = int(time.strftime("%Y%m%d"))

				getparams = {'token': self.slack_token,'pretty':'1'}
				req = requests.get('https://slack.com/api/users.list', params=getparams)
				userlist = json.loads(req.content)
				user_email = ''
				for user in userlist['members']:
					if user['id'] == sender:
						user_email = user['profile']['email']

				userid = r.hget('Users:' + user_email,'id')
				if userid == None or userid == '':
					response = 'Your email does not exist :<'
					return response

				checker = r.zrangebyscore("peopleperproject:" + projectid, userid, userid)
				if checker == None or checker == "" or checker == []:
					response = 'Your email is not recognized. Please try banana:: basecamp log <email> <"project name"> <hours> <"desc">'
					return response

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				bc.create_time_entry(desc,float(hour),int(userid),date,int(projectid),None).text
				
				return response

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp log <"project name"> <hours> <"description">'

		elif method_checker.group(2) == "getProjects":
			""" banana:: basecamp getProjects <email>"""
			regex = '(.*)\s%s\s+getProjects\s+(.*)' % self.keyword
			parser = re.match(regex,input)
			if parser:

				email = parser.group(2)
				email = email.split('|')[1][:-1]
				projectlist = []



				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				user = r.hgetall('Users:' + email)

				if user == None or user == "" or user == {}:
					response = 'Email not found!'
					return response

				userid = user['id']

				projects = r.zrange('Projects',0,-1)
				for project in projects:
					projectid = r.hget("Projects:" + project,'id')
					projectname = r.hget("Projects:" + project,'name')
					checker = r.zrangebyscore("peopleperproject:" + projectid, userid, userid)
					if checker != None and checker != "" and checker != []:
						projectlist.append(projectname)

				print projectlist
				response = "Projects of %s: \n\n" % email
				for project in projectlist:
					response = response + project + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getProjects <email>'

		elif method_checker.group(2) == "getLogs":
			""" banana:: basecamp getLogs <project name in quotation marks> <email> <yyyy-mm-dd> <yyyy-mm-dd>"""
			regex = '(.*)\s%s\s+getLogs\s+["|\'](.*)["|\']\s+(.*)\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])' % self.keyword
			parser = re.match(regex,input)
			if parser:
				
				projectid = ''
				userid = ''
				projectname = parser.group(2)
				email = parser.group(3).split('|')[1][:-1]
				date1 = parser.group(4)
				date1 = datetime(int(date1[0:4]), int(date1[5:7]), int(date1[8:10]))
				date2 = parser.group(5)
				date2 = datetime(int(date2[0:4]), int(date2[5:7]), int(date2[8:10]))
				user_time_entry = []



				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				userid = r.hget('Users:' + email,'id')

				print userid

				if userid == None or userid == "":
					response = 'Email not found!'
					return response

				pname = r.hget('Projects:' + projectname,'name')
				if pname == "" or pname == None:
					response = 'Project not found'
					return response

				projectid = r.hget('Projects:' + projectname,'id')

				userposts = r.zrangebyscore('Time_entry:' + projectid,userid,userid)
				if userposts != None and userposts != "" and userposts != []:

					for userpost in userposts:
						timedetails = r.hgetall("Time_entry:" + str(userpost))
						entrydate = timedetails['date']
						entrydate = datetime(int(entrydate[0:4]), int(entrydate[5:7]), int(entrydate[8:10]))
						if entrydate >= date1 and entrydate <= date2:
							time_instance = []
							time_instance.append(timedetails['date'])
							time_instance.append(timedetails['hours'])
							time_instance.append(timedetails['description'])
							user_time_entry.append(time_instance)

				response = "Logs of %s: \n\n" % email
				for entry in user_time_entry:
					response = response + entry[0] + "   " + entry[1] + "   " + entry[2] + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getLogs <"project name"> <email> <yyyy-mm-dd> <yyyy-mm-dd>'


		elif method_checker.group(2) == "getDistribution":
			""" banana:: basecamp getDistribution <email>"""
			regex = '(.*)\s%s\s+getDistribution\s+(.*)' % self.keyword
			parser = re.match(regex,input)
			if parser:
				print 'getdist opening!!!'
				userid = ''
				email = parser.group(2).split('|')[1][:-1]
				"""distribution = [{'name':'example','hours':'69', 'percent':100}]"""
				distribution = []
				projectlist = []

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				userid = r.hget('Users:' + email,'id')

				print userid

				if userid == None or userid == "":
					response = 'Email not found!'
					return response

				projects = r.zrange('Projects',0,-1)
				print projects


				for project in projects:
					projectid = r.hget('Projects:' + project,'id')
					projectname = r.hget('Projects:' + project,'name')

					userposts = r.zrangebyscore("Time_entry:" + projectid,userid,userid)
					if userposts != None and userposts != "" and userposts != []:

						instance = {'name':projectname,'hours':0,'percent':100}

						for userpost in userposts:
							timedetails = r.hgetall("Time_entry:" + userpost)
							instance['hours'] = instance['hours'] + float(timedetails['hours'])

						distribution.append(instance)



				response = "Logs of %s: \n\n" % email
				total_hours = 0
				for entry in distribution:
					total_hours = total_hours + entry['hours']

				for entry in distribution:
					entry['percent'] = (float(entry['hours'])/float(total_hours)) * 100

				for entry in distribution:
					response = response + entry['name'] + "   " + str(entry['percent']) + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getDistribution <email>'

		elif method_checker.group(2) == "get-logs":
			""" banana: basecamp get-logs <"Project Name">"""
			regex = '(.*)\s%s\s+get[-]logs\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response

				projectid = project['id']

				people = r.zrange('peopleperproject:' + projectid, 0, -1)

				for user in people:
					userdata = r.hgetall('Users:' + user)
					response = response + userdata['name'] + '\n'

					time_entries = r.zrangebyscore("Time_entry:" + projectid, userdata['id'], userdata['id'])
					

					csvcontent = ''

					for time_entry in time_entries:
						time_entry_data = r.hgetall("Time_entry:" + time_entry)
						#response = response + '    ' + time_entry_data['description'] + '\n'

						csvcontent = csvcontent + time_entry_data['description'] + ',' + time_entry_data['hours'] + ',' + time_entry_data['date'] + ','

						#TO ADD LATER: Put these into seperate csv files

					postparams = {"content":csvcontent}
					getparams = {"token":self.slack_token,"channels":"C02BW6E2L", "filetype":"csv", "title":userdata['name'] + ' logs'}
					req = requests.post('https://slack.com/api/files.upload',params=getparams,data=postparams)

				response = 'Logs successfully saved in csv files!'


		elif method_checker.group(2) == 'getTodoLists':
			""" banana: basecamp getTodoLists <"Project Name">"""
			regex = '(.*)\s%s\s+getTodoLists\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				todolists = r.zrange("Todo_list:" + projectid,0,-1)
				for todolist in todolists:
					todolistdata = r.hgetall("Todo_list:" + todolist)
					response = response + todolistdata['name']

					todolistid = todolistdata['id']

					response = response + "\n"

		elif method_checker.group(2) == 'getMessages':
			""" banana: basecamp getMessages <"Project Name">"""
			regex = '(.*)\s%s\s+getMessages\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				messages = r.zrange("Message:" + projectid,0,-1)
				for message in messages:
					messagedata = r.hgetall("Message:" + message)
					response = response + messagedata['body'] + '\n'


					response = response + "\n"


		elif method_checker.group(2) == 'addTodoList':
			""" banana: basecamp addTodoList <"Project Name"> <"Todo List Name">"""
			regex = '(.*)\s%s\s+addTodoList\s+["|\'](.*)["|\']\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)
				todolistname = parser.group(3)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				bc.create_todo_list(project_id=int(projectid),name=todolistname)

				response = 'Todo List created successfully'


		elif method_checker.group(2) == 'addMessages':
			""" banana: basecamp addMessages <"Project Name"> <"">"""
			regex = '(.*)\s%s\s+addMessages\s+["|\'](.*)["|\']\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			#if parser:


		else:
			response = 'Module not found!'

		return response
Beispiel #2
0
    def run(self, input, sender):
        regex = '(.*)\s%s\s+(\S*)' % self.keyword
        method_checker = re.match(regex, input)
        response = 'default'
        if (method_checker.group(2) == 'update'):
            """ banana: basecamp update"""
            self.cacheData()
            response = 'Updated!'

        elif (method_checker.group(2) == 'log'):
            """ banana:: basecamp log <email> <"project name"> <hours> <"desc">"""
            regex = '(.*)\s%s\s+log\s+(.*)\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.+[0-9])\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)
            if parser:
                projectid = ''
                projectname = parser.group(3)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = int(project['id'])

                hour = parser.group(4)
                desc = parser.group(5)
                date = int(time.strftime("%Y%m%d"))

                user_email = parser.group(2).split('|')[1][:-1]
                userid = ''

                bc = Basecamp('https://seertechnologies.basecamphq.com',
                              self.api_token)
                xml = bc.people().text
                items = ET.fromstring(xml).findall('person')
                for item in items:
                    if item.find('email-address').text == user_email:
                        userid = item.find('id').text
                        break

                if userid == '':
                    response = 'Email not recognized!'
                    return response

                projectpeople = bc.people_per_project(projectid).text
                peoplesearch = ET.fromstring(projectpeople).findall('person')
                for person in peoplesearch:
                    if int(person.find('id').text) == userid:
                        bc.create_time_entry(desc, float(hour), int(userid),
                                             date, int(projectid), None).text
                        response = 'Logged successfully!'
                        return response

                response = 'You do not belong to that project.'
                return response
            """ banana:: basecamp log <"project name"> <hours> <"desc">"""
            regex = '(.*)\s%s\s+log\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.+[0-9])\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            if parser:
                print "log 2"
                projectid = ''
                projectname = parser.group(2)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = project['id']

                hour = parser.group(3)
                desc = parser.group(4)
                date = int(time.strftime("%Y%m%d"))

                getparams = {'token': self.slack_token, 'pretty': '1'}
                req = requests.get('https://slack.com/api/users.list',
                                   params=getparams)
                userlist = json.loads(req.content)
                user_email = ''
                for user in userlist['members']:
                    if user['id'] == sender:
                        user_email = user['profile']['email']

                userid = r.hget('Users:' + user_email, 'id')
                if userid == None or userid == '':
                    response = 'Your email does not exist :<'
                    return response

                checker = r.zrangebyscore("peopleperproject:" + projectid,
                                          userid, userid)
                if checker == None or checker == "" or checker == []:
                    response = 'Your email is not recognized. Please try banana:: basecamp log <email> <"project name"> <hours> <"desc">'
                    return response

                bc = Basecamp('https://seertechnologies.basecamphq.com',
                              self.api_token)
                bc.create_time_entry(desc, float(hour), int(userid), date,
                                     int(projectid), None).text

                return response

            else:
                response = 'Wrong set of parameters. Must be banana:: basecamp log <"project name"> <hours> <"description">'

        elif method_checker.group(2) == "getProjects":
            """ banana:: basecamp getProjects <email>"""
            regex = '(.*)\s%s\s+getProjects\s+(.*)' % self.keyword
            parser = re.match(regex, input)
            if parser:

                email = parser.group(2)
                email = email.split('|')[1][:-1]
                projectlist = []

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                user = r.hgetall('Users:' + email)

                if user == None or user == "" or user == {}:
                    response = 'Email not found!'
                    return response

                userid = user['id']

                projects = r.zrange('Projects', 0, -1)
                for project in projects:
                    projectid = r.hget("Projects:" + project, 'id')
                    projectname = r.hget("Projects:" + project, 'name')
                    checker = r.zrangebyscore("peopleperproject:" + projectid,
                                              userid, userid)
                    if checker != None and checker != "" and checker != []:
                        projectlist.append(projectname)

                print projectlist
                response = "Projects of %s: \n\n" % email
                for project in projectlist:
                    response = response + project + "\n"

            else:
                response = 'Wrong set of parameters. Must be banana:: basecamp getProjects <email>'

        elif method_checker.group(2) == "getLogs":
            """ banana:: basecamp getLogs <project name in quotation marks> <email> <yyyy-mm-dd> <yyyy-mm-dd>"""
            regex = '(.*)\s%s\s+getLogs\s+["|\'](.*)["|\']\s+(.*)\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])' % self.keyword
            parser = re.match(regex, input)
            if parser:

                projectid = ''
                userid = ''
                projectname = parser.group(2)
                email = parser.group(3).split('|')[1][:-1]
                date1 = parser.group(4)
                date1 = datetime(int(date1[0:4]), int(date1[5:7]),
                                 int(date1[8:10]))
                date2 = parser.group(5)
                date2 = datetime(int(date2[0:4]), int(date2[5:7]),
                                 int(date2[8:10]))
                user_time_entry = []

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                userid = r.hget('Users:' + email, 'id')

                print userid

                if userid == None or userid == "":
                    response = 'Email not found!'
                    return response

                pname = r.hget('Projects:' + projectname, 'name')
                if pname == "" or pname == None:
                    response = 'Project not found'
                    return response

                projectid = r.hget('Projects:' + projectname, 'id')

                userposts = r.zrangebyscore('Time_entry:' + projectid, userid,
                                            userid)
                if userposts != None and userposts != "" and userposts != []:

                    for userpost in userposts:
                        timedetails = r.hgetall("Time_entry:" + str(userpost))
                        entrydate = timedetails['date']
                        entrydate = datetime(int(entrydate[0:4]),
                                             int(entrydate[5:7]),
                                             int(entrydate[8:10]))
                        if entrydate >= date1 and entrydate <= date2:
                            time_instance = []
                            time_instance.append(timedetails['date'])
                            time_instance.append(timedetails['hours'])
                            time_instance.append(timedetails['description'])
                            user_time_entry.append(time_instance)

                response = "Logs of %s: \n\n" % email
                for entry in user_time_entry:
                    response = response + entry[0] + "   " + entry[
                        1] + "   " + entry[2] + "\n"

            else:
                response = 'Wrong set of parameters. Must be banana:: basecamp getLogs <"project name"> <email> <yyyy-mm-dd> <yyyy-mm-dd>'

        elif method_checker.group(2) == "getDistribution":
            """ banana:: basecamp getDistribution <email>"""
            regex = '(.*)\s%s\s+getDistribution\s+(.*)' % self.keyword
            parser = re.match(regex, input)
            if parser:
                print 'getdist opening!!!'
                userid = ''
                email = parser.group(2).split('|')[1][:-1]
                """distribution = [{'name':'example','hours':'69', 'percent':100}]"""
                distribution = []
                projectlist = []

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                userid = r.hget('Users:' + email, 'id')

                print userid

                if userid == None or userid == "":
                    response = 'Email not found!'
                    return response

                projects = r.zrange('Projects', 0, -1)
                print projects

                for project in projects:
                    projectid = r.hget('Projects:' + project, 'id')
                    projectname = r.hget('Projects:' + project, 'name')

                    userposts = r.zrangebyscore("Time_entry:" + projectid,
                                                userid, userid)
                    if userposts != None and userposts != "" and userposts != []:

                        instance = {
                            'name': projectname,
                            'hours': 0,
                            'percent': 100
                        }

                        for userpost in userposts:
                            timedetails = r.hgetall("Time_entry:" + userpost)
                            instance['hours'] = instance['hours'] + float(
                                timedetails['hours'])

                        distribution.append(instance)

                response = "Logs of %s: \n\n" % email
                total_hours = 0
                for entry in distribution:
                    total_hours = total_hours + entry['hours']

                for entry in distribution:
                    entry['percent'] = (float(entry['hours']) /
                                        float(total_hours)) * 100

                for entry in distribution:
                    response = response + entry['name'] + "   " + str(
                        entry['percent']) + "\n"

            else:
                response = 'Wrong set of parameters. Must be banana:: basecamp getDistribution <email>'

        elif method_checker.group(2) == "get-logs":
            """ banana: basecamp get-logs <"Project Name">"""
            regex = '(.*)\s%s\s+get[-]logs\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            response = ''

            if parser:
                projectname = parser.group(2)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = project['id']

                people = r.zrange('peopleperproject:' + projectid, 0, -1)

                for user in people:
                    userdata = r.hgetall('Users:' + user)
                    response = response + userdata['name'] + '\n'

                    time_entries = r.zrangebyscore("Time_entry:" + projectid,
                                                   userdata['id'],
                                                   userdata['id'])

                    csvcontent = ''

                    for time_entry in time_entries:
                        time_entry_data = r.hgetall("Time_entry:" + time_entry)
                        #response = response + '    ' + time_entry_data['description'] + '\n'

                        csvcontent = csvcontent + time_entry_data[
                            'description'] + ',' + time_entry_data[
                                'hours'] + ',' + time_entry_data['date'] + ','

                        #TO ADD LATER: Put these into seperate csv files

                    postparams = {"content": csvcontent}
                    getparams = {
                        "token": self.slack_token,
                        "channels": "C02BW6E2L",
                        "filetype": "csv",
                        "title": userdata['name'] + ' logs'
                    }
                    req = requests.post('https://slack.com/api/files.upload',
                                        params=getparams,
                                        data=postparams)

                response = 'Logs successfully saved in csv files!'

        elif method_checker.group(2) == 'getTodoLists':
            """ banana: basecamp getTodoLists <"Project Name">"""
            regex = '(.*)\s%s\s+getTodoLists\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            response = ''

            if parser:
                projectname = parser.group(2)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = project['id']

                todolists = r.zrange("Todo_list:" + projectid, 0, -1)
                for todolist in todolists:
                    todolistdata = r.hgetall("Todo_list:" + todolist)
                    response = response + todolistdata['name']

                    todolistid = todolistdata['id']

                    response = response + "\n"

        elif method_checker.group(2) == 'getMessages':
            """ banana: basecamp getMessages <"Project Name">"""
            regex = '(.*)\s%s\s+getMessages\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            response = ''

            if parser:
                projectname = parser.group(2)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = project['id']

                messages = r.zrange("Message:" + projectid, 0, -1)
                for message in messages:
                    messagedata = r.hgetall("Message:" + message)
                    response = response + messagedata['body'] + '\n'

                    response = response + "\n"

        elif method_checker.group(2) == 'addTodoList':
            """ banana: basecamp addTodoList <"Project Name"> <"Todo List Name">"""
            regex = '(.*)\s%s\s+addTodoList\s+["|\'](.*)["|\']\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            response = ''

            if parser:
                projectname = parser.group(2)
                todolistname = parser.group(3)

                r = redis.StrictRedis(host='localhost', port=6379, db=0)
                project = r.hgetall("Projects:" + projectname)
                if project == None or project == [] or project == {}:
                    response = 'Project not found'
                    return response

                projectid = project['id']

                bc = Basecamp('https://seertechnologies.basecamphq.com',
                              self.api_token)
                bc.create_todo_list(project_id=int(projectid),
                                    name=todolistname)

                response = 'Todo List created successfully'

        elif method_checker.group(2) == 'addMessages':
            """ banana: basecamp addMessages <"Project Name"> <"">"""
            regex = '(.*)\s%s\s+addMessages\s+["|\'](.*)["|\']\s+["|\'](.*)["|\']' % self.keyword
            parser = re.match(regex, input)

            response = ''

            #if parser:

        else:
            response = 'Module not found!'

        return response
Beispiel #3
0
	def run(self,input,sender,channel):
		regex = '(.*)\s%s\s+(\S*)' % self.keyword
		method_checker = re.match(regex,input)
		response = 'default'
		if(method_checker.group(2) == 'update'):
			""" banana: basecamp update"""
			self.cacheData()
			response = 'Updated!'

		elif(method_checker.group(2) == 'Cron'):
			""" banana: basecamp Cron"""
			
			sched = Scheduler()
			
			sched.add_cron_job(self.cacheData,hour=2,minute=0)

			sched.start()

			response = 'Started cron'

		elif(method_checker.group(2) == 'log'):

			response = 'Logging turned off'
			return response

			""" banana:: basecamp log <email> <"project name"> <hours> <"desc">"""
			regex = '(.*)\s%s\s+log\s+(.*)\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.?[0-9]+)\s+["|\'](.*)["|\']((\s+)([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9]))?' % self.keyword
			parser = re.match(regex,input)
			if parser:
				projectid = ''
				projectname = parser.group(3)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
				
				projectid = int(project['id'])

				hour = parser.group(4)
				desc = parser.group(5)
				date = ''
				inputdate = parser.group(6)
				if inputdate == "" or inputdate is None:
					date = int(time.strftime("%Y%m%d"))

				else:
					date = int(inputdate[1:5] + inputdate[6:8] + inputdate[9:11])

				user_email = parser.group(2)
				userid = ''

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				xml = bc.people().text
				items = ET.fromstring(xml).findall('person')
				for item in items:
					if item.find('email-address').text == user_email:
						userid = item.find('id').text
						break
				
				if userid == '':
					response = 'Email not recognized!'
					return response

				projectpeople = bc.people_per_project(projectid).text
				peoplesearch = ET.fromstring(projectpeople).findall('person')
				for person in peoplesearch:
					if person.find('id').text == userid:
						bc.create_time_entry(desc,float(hour),int(userid),date,int(projectid),None).text
						response = 'Logged successfully!'

						totalLogs = float(r.hget('Users:' + userid, 'BananaLogsForTheWeek'))
						totalLogs = totalLogs + float(hour)
						r.hset('Users:' + userid, 'BananaLogsForTheWeek', totalLogs)

						bananaUsage = r.hget('Users:' + userid, 'BananaUsageForTheWeek')
						dateToday = datetime.now().weekday()
						if dateToday == len(bananaUsage):
							bananaUsage = bananaUsage + 'B'
							r.hset('Users:' + userid, 'BananaUsageForTheWeek',bananaUsage)

						return response

				response = 'You do not belong to that project.'
				return response


			""" banana:: basecamp log <"project name"> <hours> <"desc">"""
			regex = '(.*)\s%s\s+log\s+["|\'](.*)["|\']\s+([0-9]*[0-9]?\.?[0-9]+)\s+["|\'](.*)["|\']((\s+)([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9]))?' % self.keyword
			parser = re.match(regex,input)
			
			if parser:
				projectid = ''
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
				
				projectid = project['id']

				hour = parser.group(3)
				desc = parser.group(4)

				date = ''
				inputdate = parser.group(5)
				if inputdate == "" or inputdate is None:
					date = int(time.strftime("%Y%m%d"))

				else:
					date = int(inputdate[1:5] + inputdate[6:8] + inputdate[9:11])


				getparams = {'token': self.slack_token,'pretty':'1'}
				req = requests.get('https://slack.com/api/users.list', params=getparams, verify=False)
				userlist = json.loads(req.content)
				user_email = ''
				for user in userlist['members']:
					if user['id'] == sender:
						user_email = user['profile']['email']

				userid = r.hget('Users:' + user_email,'id')
				if userid == None or userid == '':
					response = 'Your email does not exist :<'
					return response

				checker = r.zrangebyscore("peopleperproject:" + projectid, userid, userid)
				if checker == None or checker == "" or checker == []:
					response = 'Your email is not recognized. Please try banana:: basecamp log <email> <"project name"> <hours> <"desc">'
					return response

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				bc.create_time_entry(desc,float(hour),int(userid),date,int(projectid),None).text
				
				totalLogs = float(r.hget('Users:' + userid, 'BananaLogsForTheWeek'))
				totalLogs = totalLogs + float(hour)
				r.hset('Users:' + userid, 'BananaLogsForTheWeek', totalLogs)

				bananaUsage = r.hget('Users:' + userid, 'BananaUsageForTheWeek')
				dateToday = datetime.now().weekday()
				if dateToday == len(bananaUsage):
					bananaUsage = bananaUsage + 'B'
					r.hset('Users:' + userid, 'BananaUsageForTheWeek',bananaUsage)

				response = 'successfully logged!'
				return response

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp log <"project name"> <hours> <"description">'

		elif method_checker.group(2) == "getProjects":
			""" banana:: basecamp getProjects <email>"""
			regex = '(.*)\s%s\s+getProjects\s+(.*)' % self.keyword
			parser = re.match(regex,input)
			if parser:

				email = parser.group(2)
				projectlist = []



				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				user = r.hgetall('Users:' + email)

				if user == None or user == "" or user == {}:
					response = 'Email not found!'
					return response

				userid = user['id']

				projects = r.zrange('Projects',0,-1)
				for project in projects:
					projectid = r.hget("Projects:" + project,'id')
					projectname = r.hget("Projects:" + project,'name')
					checker = r.zrangebyscore("peopleperproject:" + projectid, userid, userid)
					if checker != None and checker != "" and checker != []:
						projectlist.append(projectname)

				response = "Projects of %s: \n\n" % email
				for project in projectlist:
					response = response + project + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getProjects <email>'

		elif method_checker.group(2) == "getLogs":
			""" banana:: basecamp getLogs <project name in quotation marks> <email> <yyyy-mm-dd> <yyyy-mm-dd>"""
			regex = '(.*)\s%s\s+getLogs\s+["|\'](.*)["|\']\s+(.*)\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])\s+([0-9][0-9][0-9][0-9][-][0-9][0-9][-][0-9][0-9])' % self.keyword
			parser = re.match(regex,input)
			if parser:
				
				projectid = ''
				userid = ''
				projectname = parser.group(2)
				email = parser.group(3)
				date1 = parser.group(4)
				date1 = datetime(int(date1[0:4]), int(date1[5:7]), int(date1[8:10]))
				date2 = parser.group(5)
				date2 = datetime(int(date2[0:4]), int(date2[5:7]), int(date2[8:10]))
				user_time_entry = []



				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				userid = r.hget('Users:' + email,'id')

				if userid == None or userid == "":
					response = 'Email not found!'
					return response

				pname = r.hget('Projects:' + projectname,'name')
				if pname == "" or pname == None:
					response = 'Project not found'
					return response

				projectid = r.hget('Projects:' + projectname,'id')

				userposts = r.zrangebyscore('Time_entry:' + projectid,userid,userid)
				if userposts != None and userposts != "" and userposts != []:

					for userpost in userposts:
						timedetails = r.hgetall("Time_entry:" + str(userpost))
						entrydate = timedetails['date']
						entrydate = datetime(int(entrydate[0:4]), int(entrydate[5:7]), int(entrydate[8:10]))
						if entrydate >= date1 and entrydate <= date2:
							time_instance = []
							time_instance.append(timedetails['date'])
							time_instance.append(timedetails['hours'])
							time_instance.append(timedetails['description'])
							user_time_entry.append(time_instance)

				response = "Logs of %s: \n\n" % email
				for entry in user_time_entry:
					response = response + entry[0] + "   " + entry[1] + "   " + entry[2] + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getLogs <"project name"> <email> <yyyy-mm-dd> <yyyy-mm-dd>'


		elif method_checker.group(2) == "getDistribution":
			""" banana:: basecamp getDistribution <email>"""
			regex = '(.*)\s%s\s+getDistribution\s+(.*)' % self.keyword
			parser = re.match(regex,input)
			if parser:
				userid = ''
				email = parser.group(2)
				"""distribution = [{'name':'example','hours':'69', 'percent':100}]"""
				distribution = []
				projectlist = []

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				userid = r.hget('Users:' + email,'id')

				if userid == None or userid == "":
					response = 'Email not found!'
					return response

				projects = r.zrange('Projects',0,-1)

				for project in projects:
					projectid = r.hget('Projects:' + project,'id')
					projectname = r.hget('Projects:' + project,'name')

					userposts = r.zrangebyscore("Time_entry:" + projectid,userid,userid)
					if userposts != None and userposts != "" and userposts != []:

						instance = {'name':projectname,'hours':0,'percent':100}

						for userpost in userposts:
							timedetails = r.hgetall("Time_entry:" + userpost)
							instance['hours'] = instance['hours'] + float(timedetails['hours'])

						distribution.append(instance)



				response = "Logs of %s: \n\n" % email
				total_hours = 0
				for entry in distribution:
					total_hours = total_hours + entry['hours']

				for entry in distribution:
					entry['percent'] = round((float(entry['hours'])/float(total_hours)) * 100,2)

				for entry in distribution:
					response = response + entry['name'] + "   " + str(entry['percent']) + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getDistribution <email>'

		elif method_checker.group(2) == "get-logs":
			""" banana: basecamp get-logs <"Project Name">"""
			regex = '(.*)\s%s\s+get[-]logs\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response

				projectid = project['id']

				people = r.zrange('peopleperproject:' + projectid, 0, -1)

				for user in people:
					userdata = r.hgetall('Users:' + user)
					response = response + userdata['name'] + '\n'

					time_entries = r.zrangebyscore("Time_entry:" + projectid, userdata['id'], userdata['id'])
					

					csvcontent = ''

					for time_entry in time_entries:
						time_entry_data = r.hgetall("Time_entry:" + time_entry)
						#response = response + '    ' + time_entry_data['description'] + '\n'

						csvcontent = csvcontent + time_entry_data['description'] + ',' + time_entry_data['hours'] + ',' + time_entry_data['date'] + ','

						#TO ADD LATER: Put these into seperate csv files

					postparams = {"content":csvcontent}
					getparams = {"token":self.slack_token,"channels":"C02BW6E2L", "filetype":"csv", "title":userdata['name'] + ' logs'}
					req = requests.post('https://slack.com/api/files.upload',params=getparams,data=postparams)

				response = 'Logs successfully saved in csv files!'


		elif method_checker.group(2) == 'getTodoLists':
			""" banana: basecamp getTodoLists <"Project Name">"""
			regex = '(.*)\s%s\s+getTodoLists\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				todolists = r.zrange("Todo_list:" + projectid,0,-1)
				for todolist in todolists:
					todolistdata = r.hgetall("Todo_list:" + todolist)
					response = response + todolistdata['name']

					todolistid = todolistdata['id']

					response = response + "\n"

		elif method_checker.group(2) == 'getMessages':
			""" banana: basecamp getMessages <"Project Name">"""
			regex = '(.*)\s%s\s+getMessages\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				messages = r.zrange("Message:" + projectid,0,-1)
				for message in messages:
					messagedata = r.hgetall("Message:" + message)

					parser = MessageParser()
					parser.feed(messagedata['body'])

					response = response + parser.output + '\n'
					response = response + "\n"

					parser.close()

		elif method_checker.group(2) == 'addTodoList':
			""" banana: basecamp addTodoList <"Project Name"> <"Todo List Name">"""
			regex = '(.*)\s%s\s+addTodoList\s+["|\'](.*)["|\']\s+["|\'](.*)["|\']' % self.keyword
			parser = re.match(regex,input)

			response = ''

			if parser:
				projectname = parser.group(2)
				todolistname = parser.group(3)

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				project = r.hgetall("Projects:" + projectname)
				if project == None or project == [] or project == {}:
					response = 'Project not found'
					return response
					
				projectid = project['id']

				bc = Basecamp('https://seertechnologies.basecamphq.com', self.api_token)
				bc.create_todo_list(project_id=int(projectid),name=todolistname)

				response = 'Todo List created successfully'

		elif method_checker.group(2) == "getDistributionLastMonth":
			""" banana:: basecamp getDistributionLastMonth <email>"""
			regex = '(.*)\s%s\s+getDistributionLastMonth\s+(.*)' % self.keyword
			parser = re.match(regex,input)
			if parser:
				userid = ''
				email = parser.group(2)
				"""distribution = [{'name':'example','hours':'69', 'percent':100}]"""
				distribution = []
				projectlist = []

				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				userid = r.hget('Users:' + email,'id')

				if userid == None or userid == "":
					response = 'Email not found!'
					return response

				projects = r.zrange('Projects',0,-1)

				for project in projects:
					projectid = r.hget('Projects:' + project,'id')
					projectname = r.hget('Projects:' + project,'name')

					userposts = r.zrangebyscore("Time_entry:" + projectid,userid,userid)
					if userposts != None and userposts != "" and userposts != []:

						instance = {'name':projectname,'hours':0,'percent':100}

						for userpost in userposts:
							timedetails = r.hgetall("Time_entry:" + userpost)

							entrydate = timedetails['date']
							entrydate = datetime(int(entrydate[0:4]), int(entrydate[5:7]), int(entrydate[8:10]))
							datetoday = datetime.now()

							if (entrydate.month+1 == datetoday.month and entrydate.year == datetoday.year) or (entrydate.month == 12 and datetoday.month == 1 and entrydate.year+1 == datetoday.year):
								instance['hours'] = instance['hours'] + float(timedetails['hours'])

						distribution.append(instance)



				response = "Logs of %s: \n\n" % email
				total_hours = 0
				for entry in distribution:
					total_hours = total_hours + entry['hours']

				for entry in distribution:
					entry['percent'] = round((float(entry['hours'])/float(total_hours)) * 100,2)

				for entry in distribution:
					response = response + entry['name'] + "   " + str(entry['percent']) + "\n"

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getDistributionLastMonth <email>'

		elif method_checker.group(2) == "getStarbucks":
			""" banana:: basecamp getStarbucks"""
			regex = '(.*)\s%s\s+getStarbucks' % self.keyword
			parser = re.match(regex,input)
			if parser:

				response = 'People who logged at least 32 hours using banana last week: \n'
				r = redis.StrictRedis(host='localhost',port=6379,db=0)
				users = r.zrange('Users',0,-1)
				for user in users:
					totalLogs = r.hget('Users:' + user, 'totalLogsLastWeek')
					bananalogs = r.hget('Users:' + user, 'totalLogsLastWeek')
					bananaUsage = r.hget('Users:' + user, 'totalLogsLastWeek')

			else:
				response = 'Wrong set of parameters. Must be banana:: basecamp getProjects <email>'

		else:
			response = 'Module not found!'

		return response