client

#!/usr/bin/python3.5

# import argparse
# parser = argparse.ArgumentParser(description='Ayuda client app')
# parser.add_argument('listar_dispositivos', type=str,help='listar todos los dispositivos del usb')
# parser.add_argument('nombrar_dispositivo', type=str,help='darle un nombre a los dispositivos')
# parser.add_argument('leer_archivo', type=str,help='obtener contenido de un archivo')
# parser.add_argument('escribir_archivo', type=str,help='anadir contenido a un archivo del usb')
# args = parser.parse_args()
# print(args.accumulate(args.integers))

import sys
import requests
import json
from tabulate import tabulate
import os
# print ('Argument List:', str(sys.argv))
ayuda = """
		Ejecute para ayuda:
		
		\033[92m./client --help\033[0m
	"""
def is_json(myjson):
  try:
    json_object = json.loads(myjson)
  except ValueError:
    return False
  return True
opciones_string = """
Uso: python3 client.py PUERTO [solicitud] [argumentos...]
	
	\033[92mpuerto:\033[0m
		PUERTO                       nombre del puerto en que se esta ejecutando el servidor, Ej: 8888

	\033[92msolicitud:\033[0m
		listar_dispositivos
		nombrar_dispositivo
		leer_archivo
		escribir_archivo
	
	\033[1mArgumentos listar_dispositivos:\033[0m
		Listar todos los dispositivos conectado al pc

		\033[94mEjemplo: ./client 8888 listar_dispositivos\033[0m
	
	\033[1mArgumentos nombrar_dispositivo:\033[0m
		nodo                         nodo del localizacion del usb, Ej: /dev/sdb1
		nombre                       nombre del usb que se desea dar, Ej: mi_usb

		\033[94mEjemplo: ./client 8888 nombrar_dispositivo /dev/sdb1 mi_usb\033[0m

	
	\033[1mArgumento leer_archivo:\033[0m
		nombre                       nombre del usb previamente establecido
		nombre_archivo               nombre del archivo relativa al usb, Ej: mi_archivo.txt  
		
		\033[94mEjemplo: ./client 8888 leer_archivo mi_usb mi_archivo.txt\033[0m
	
	
	\033[1mArgumento escribir_archivo:\033[0m
		nombre 		             nombre del usb previamente establecido
		nombre_archivo               nombre del archivo relativa al usb, Ej: mi_archivo.txt
		tamano_contenido             tamano del contenido que se desea guardar
		contenido                    direccion del archivo que contiene lo que se desea enviar, Ej: /home/user/contenido.txt
		
		\033[94mEjemplo: ./client 8888 escribir_archivo mi_usb mi_archivo.txt 50 /home/user/contenido.txt\033[0m
"""

def listar_dispositivos():
	r = requests.get("http://localhost:{}/listar_dispositivos".format(int(sys.argv[1])), data=None)
	# print(r.text)
	respuesta = json.loads(r.text)
	resp = []
	if (respuesta['status'] == 0):	
		for respuesta in respuesta['dispositivos']:
			tmp = []
			tmp.append( respuesta['nombre'])
			tmp.append( respuesta['montaje'])
			tmp.append( respuesta['id'])
			tmp.append( respuesta['nodo'])
			resp.append(tmp)
		print (tabulate(resp, headers=["nombre","montaje", "id", "nodo"], tablefmt="grid"))
	else:
		print("\033[0;31m" +respuesta['str_error'] + "\033[0m")

def nombrar_dispositivo():
	if len(sys.argv) == 5:
		req ="{" +  '"solicitud": "nombrar_dispositivo","nodo": "{}", "nombre": "{}"'.format(sys.argv[3].replace('/', '\\/'),sys.argv[4]) + "}" 
		r = requests.post("http://localhost:{}/nombrar_dispositivo".format(int(sys.argv[1])), data={"nombrar_dispositivo": req})
		respuesta = json.loads(r.text)
		# print(r.text)
		if (respuesta['status'] == 0):
			resp = []
			tmp = []
			tmp.append( respuesta['nombre'])
			tmp.append( respuesta['nodo'])
			resp.append(tmp)
			print (tabulate(resp, headers=["nombre", "nodo"], tablefmt="grid"))
		else:
			print("\033[0;31m" +respuesta['str_error'] + "\033[0m")
		
	else:
		print("El numero de argumentos no es valido")
		print(ayuda)

def leer_archivo():
	if len(sys.argv) == 5:
		req ="{" +  '"solicitud": "leer_archivo", "nombre": "{}","nombre_archivo": "{}"'.format(sys.argv[3],sys.argv[4]) + "}"
		r = requests.post("http://localhost:{}/leer_archivo".format(int(sys.argv[1])), data={"leer_archivo": req})
		# print(r.text)
		respuesta = json.loads(r.text)
		# "nombre": "mi_usb", "nombre_del_archivo": "archivo.txt", "contenido": "sdadad\na\nsd\nsd\nasd",
		if (respuesta['status'] == 0):
			# print(r.text)
			resp = []
			tmp = []
			tmp.append( respuesta['nombre'])
			tmp.append( respuesta['nombre_archivo'])
			resp.append(tmp)
			print (tabulate(resp, headers=["nombre", "nombre_del_archivo"], tablefmt="grid"))
			print("\n\033[92mContenido\033[0m\n")
			print(respuesta['contenido'])
			print("\n")
		else:
			print("\033[0;31m" +respuesta['str_error'] + "\033[0m")
		# print(r.text)
	else:
		print("El numero de argumentos no es valido")
		print(ayuda)

def escribir_archivo():
	# TODO: leer archivo para escribirlo
	# print(sys.argv[6]

	if len(sys.argv) == 7:
		try:
			es_digito = sys.argv[5].isdigit()
			if es_digito:
				with open(sys.argv[6], 'r') as myfile:
					bytes = os.path.getsize(sys.argv[6])
					print("El tamano contenido es: {}".format(bytes))
					if bytes!= int(sys.argv[5]):
						print("El tamano del contenido no es correcto")
					else:
						data = myfile.read().encode('utf8')
						data = str(data)
						data = data[2:-1]
						req ="{" +  '"solicitud": "escribir_archivo","nombre": "{}", "nombre_archivo": "{}", "tamano_contenido": {}, "contenido": "{}"'.format(sys.argv[3],sys.argv[4],bytes, data) + "}"
						r = requests.post("http://localhost:{}/escribir_archivo".format(int(sys.argv[1])), data={"escribir_archivo": req})
						# print(r.text)
						respuesta = json.loads(r.text)
						if (respuesta['status'] == 0):
							tmp = []
							resp = []
							tmp.append(respuesta['solicitud'])
							tmp.append(respuesta['nombre'])
							resp.append(tmp)
							print (tabulate(resp, headers=["solicitud", "nombre"], tablefmt="grid"))
						else:
							print("\033[0;31m" +respuesta['str_error'] + "\033[0m")
			else:
				print("El tamano contenido no es un numero")

		except:
			print("El archivo no existe")
	else:
		print("El numero de argumentos no es valido")
		print(ayuda)

def opciones():
	try:
		r = requests.get("http://localhost:{}".format(int(sys.argv[1])), data=None)
		if (str(sys.argv[2]) == 'listar_dispositivos'):
			listar_dispositivos()
		elif (str(sys.argv[2]) == 'nombrar_dispositivo'):
			nombrar_dispositivo()
		elif (str(sys.argv[2]) == 'leer_archivo'):
			leer_archivo()
		elif (str(sys.argv[2]) == 'escribir_archivo'):
			escribir_archivo()
		else:
			print("La solicitud no es valida")
	except:
		print("El server no esta escuchando en este puerto")

def main():
	estado = True
	if (len(sys.argv) > 1):
		if (str(sys.argv[1]) == '--help'):

			print(opciones_string)
		
		if (str(sys.argv[1]) != '--help'):
			try:
				if (len(sys.argv) == 2):
					print("No hay solicitud")
				else:
					opciones()
				pass
			except:
				print("El puerto no es un numero")
	else:
		print (ayuda)

if __name__ == "__main__":
    main()




"""
GET
/listar_dispositivo

*response
{
    solicitud: 'listar_dispositivos',
    dispositivos: [{
        nombre: '',
        id: 'vendor:device',
        montaje: '/home',
        nodo: '/dev/'
    }],
    status: 0,
    str_error: ''
}
"""

"""
POST
/nombrar_dispositivo

*request
{
    solicitud: 'nombrar_dispositivo',
    nodo: '/dev',
    nombre: ''
}

*response
{
    solicitud: 'nombrar_dispositivo',
    nombre: '',
    nodo: '/dev/'
    status: 0,
    str_error: ''
    status: 0,
}
"""

"""
GET
/leer_archivo

*request
{
    solicitud: 'leer_archivo',
    nombre: '',
    nombre_archivo: ''
}

*response
{
    solicitud: 'leer_archivo',
    nombre: '',
    nombre_archivo: '',
    contenido: ''
    status: 0,
    str_error: ''
}
"""

"""
POST
/escribir_archivo

*request
{
    solicitud: 'escribir_archivo',
    nombre: '',
    nombre_archivo: '',
    tamano_contenido: '',
    contenido: ''
}

*response
{
    solicitud: 'escribir_archivo',
    nombre: '',
    nombre_archivo: '',
    status: 0,
    str_error: ''
}
"""