# -*- coding: utf-8 -*-

import os
import shutil
import sqlite3

class PasswordManager:
	
	PASSWORD_FILE="/net/var/lib/n4d/n4d.sqlite"
	LOG_PATH="/net/var/lib/n4d"
	
	def __init__(self):
			
		if not os.path.exists(PasswordManager.PASSWORD_FILE):
			if not os.path.exists(PasswordManager.LOG_PATH):
				os.makedirs(PasswordManager.LOG_PATH)
			self.create_database()
			prevmask = os.umask(0)
			os.chmod(PasswordManager.PASSWORD_FILE,0640)
			os.umask(prevmask)
		
		
	#def init
	
	def create_database(self):
		conn = sqlite3.connect(PasswordManager.PASSWORD_FILE)
		cursor = conn.cursor()
		cursor.execute('create table password (cn text, sn text, uid text, passwd text)')
		conn.commit()
		cursor.close()
		conn.close()
	
	def add_password(self,user_name,cn,sn,password):
		conn = sqlite3.connect(PasswordManager.PASSWORD_FILE)
		cursor = conn.cursor()
		cursor.execute('select * from password where uid = "'+user_name+'"')
		result = cursor.fetchall()
		if len(result) > 0 :
			# user on database
			cursor.execute('update password set passwd = "'+password+'" where uid = "'+user_name+'"' )
		else:
			cursor.execute('insert into password (cn,sn,uid,passwd) values ("'+cn+'","'+sn+'","'+user_name+'","'+password+'")')
			
		conn.commit()
		cursor.close()
		conn.close()
	
		
	#def add_password
	
	def remove_password(self,user_name):
		conn = sqlite3.connect(PasswordManager.PASSWORD_FILE)
		cursor = conn.cursor()
		cursor.execute('delete from password where uid = "'+user_name+'"')
		conn.commit()
		cursor.close()
		conn.close()
	
	#def remove_password
	
	def get_passwords(self):
		
		list = []
		conn = sqlite3.connect(PasswordManager.PASSWORD_FILE)
		cursor = conn.cursor()
		cursor.execute('select cn,sn,uid,passwd from password')
		result = cursor.fetchall()
		for user in result :
			a = {}
			a['cn'] = user[0].encode('utf-8')
			a['sn'] = user[1].encode('utf-8')
			a['uid'] = user[2]
			a['passwd'] = user[3]
			list.append(a)
		conn.close()
		return list
		
	#def get_passwords
	
	
	
#class PasswordManager

if __name__=="__main__":
	
	pm=PasswordManager()
	pm.add_password("test","password2")
	pm.add_password("test2","password2")
	pm.remove_password("raul")