# -*- coding: utf-8  -*-
"""
Vacía la zona de pruebas con menú
"""

import wikipedia

# parámetros:
mensaje = u'<!--No borres este mensaje-->{{Titular-zdp}}<!--Haz las pruebas debajo, gracias-->\n\n== Las pruebas en esta sección =='
resumen = u'Bot: Limpieza de la zona de pruebas ([[WP:ZP]])'
nombre_zona_pruebas = u'Wikipedia:Zona de pruebas'

def limpiar(pagina):
	zn=wikipedia.Page(wikipedia.getSite(), pagina)
	print 'Limpiado %s' % pagina
	zn.put(mensaje,resumen)

if __name__ == '__main__':	
	salir = 0
	while (salir==0):
		print u"\nIntroducir el número de zona a limpiar (0=Todas)."
		print u"Si no se indica ninguna se limpiará solo la primera (9=Salir)."
		opcion=raw_input('> ')

		if(opcion=='0'):
			limpiar(nombre_zona_pruebas)
			for i in range(2, 6):
				limpiar(nombre_zona_pruebas+'/'+str(i))
		elif(opcion=='1'):
			limpiar(nombre_zona_pruebas)
		elif(opcion in ['2', '3', '4', '5']):
			limpiar(nombre_zona_pruebas+'/'+opcion)
		elif(opcion=='9'):
			break
		else:
			limpiar(nombre_zona_pruebas)