import tempfile
import shutil
import os
import subprocess
import time

class MysqlManager:
	
	
	def __init__(self):
		
		pass
		
	#def init
	
	def get_time(self):
		
		return get_backup_name("MysqlManager")
		
	#def get_time
	
	
	def backup(self,dir="/backup/"):
		
		try:
		
			file_path=dir+"/"+self.get_time()
			tmp_dir=tempfile.mkdtemp()
			
			ret=os.system("mysql_root_passwd -g")
			
			if ret!=0:
				os.system("mysql_root_passwd -i")
			
			cmd="mysqldump -u root -p$(mysql_root_passwd -g) --routines --all-databases --flush-privileges > %s/backup.sql"%(tmp_dir)
			os.system(cmd)
			
			if os.path.exists(tmp_dir+"/backup.sql"):
				tar=tarfile.open(file_path,"w:gz")
				tar.add(tmp_dir+"/backup.sql",arcname="backup.sql")
				if os.path.exists("/root/.my.cnf"):
					tar.add("/root/.my.cnf",arcname="my.cnf")
				tar.close()
				
				os.system("chmod 600 %s"%file_path)
				
				return [True,file_path]
			
		except Exception as e:
			return [False,str(e)]
			
		return [False,"Mysqldump failed"]
		
	#def test
	
	def restore(self,file_path=None):

		if file_path==None:
			for f in sorted(os.listdir("/backup"),reverse=True):
				if "MysqlManager" in f:
					file_path="/backup/"+f
					break

		if file_path==None:
			return [False,"No backup file found"]
			
		try:

			if os.path.exists(file_path):
				
				tmp_dir=tempfile.mkdtemp()
				tar=tarfile.open(file_path,"r")
				tar.extractall(tmp_dir)
				tar.close()
				#os.system("chmod 600 %s"%file_path)
				
				if os.path.exists(tmp_dir+"/my.cnf"):
					shutil.copy(tmp_dir+"/my.cnf","/root/.my.cnf")
					os.system("chmod 600 /root/.my.cnf")
					os.system("mysql_root_passwd -i")
					
				if os.path.exists(tmp_dir+"/backup.sql"):
					cmd="mysql -u root -p$(mysql_root_passwd -g) < %s"%(tmp_dir+"/backup.sql")
					os.system(cmd)
					
				
				return [True,""]
				
		except Exception as e:
			print e
			return [False,file_path + ": " + str(e)]
		
	#def test


	
	
#class CupsManager

if __name__=="__main__":
	
	m=MysqlManager()
		
	m.backup()