# -*- coding: utf-8 -*-
import PAM
class PamManager:
	def __init__(self,module="login"):
		self.user = None
		self.passwd = None
		self.auth = PAM.pam()
		self.auth.start(module)
	
	def authentication(self,user,passwd):
		
		def pam_conv(auth, query_list, userData):
			resp = []
			for i in range(len(query_list)):
				query, type = query_list[i]
				if type == PAM.PAM_PROMPT_ECHO_ON :
					resp.append((user),0)
				if type == PAM.PAM_PROMPT_ECHO_OFF:
					resp.append((passwd, 0))
				elif type == PAM.PAM_ERROR_MSG or type == PAM.PAM_TEXT_INFO:
					#print query
					resp.append(('', 0))
				else:
					return None
			return resp
		self.auth.set_item(PAM.PAM_USER, user)
		self.auth.set_item(PAM.PAM_CONV,pam_conv)
		try:
			self.auth.authenticate()
			self.auth.acct_mgmt()
			return True
		except Exception as e:
			#f = open('/var/log/n4d/pam','a')
			#f.write(e.message + "\n")
			#f.close()
			return False