levelGenerator.py

import os,time
from characters import *
from level import *
from getch import *

class LevelGenerator:

	def main():
		os.system('clear')
		print("1. Edit Existing")
		print("2. Start from Scratch")
		ch = getchS()
		filename = "level"
		if ch is '1':
			while True: 
				try:
					print()
					print("All levels :")
					os.system("ls ./levels/ | grep -oP '^((?!Enemy).)*$' | sed -e 's/\.txt$//'")
					print()
					print("Enter Level Name : ",end='')
					filename = input()
					level = Level(0,0,"./levels/"+filename+".txt")
					break
				except FileNotFoundError:
					print("Level Does not exist! Try again")
		else:
			level = Level(0,0,"./designs/"+filename+".txt")

		enemies = []
		enemiesstr = []

		try:
			file = open("./levels/"+filename+"Enemy.txt","r")
			for line in file:
				enemiesstr.append(line)
				goomba = eval(line)
				goomba.cannotCross=['T','|','/','\\','`']
				enemies.append(goomba)
		except:
			pass

		rules = Element(0,0)
		while True:
			rules.design = ["|   b : brick   |    c : coin   |   s : spring  |   t : tunnel  |    p : pit   |","| m : coinbrick |   h : health  |    g : gun    |   e : enemy   |    q : done  |"]
			level.printOnTop(*enemies,rules)

			choice = getchS()
			if choice is 'b':
				obj = Character(0,level.pos,2,"./designs/brick.txt")
				obj.cannotCross=['T','/','\\','`','-']
			elif choice is 'c':
				obj = Character(0,level.pos,2,"./designs/coin.txt")
				obj.cannotCross=['T','|','/','\\','`','-']
			elif choice is 's':
				obj = Character(0,level.pos,2,"./designs/spring.txt")
				obj.cannotCross=['|','/','\\','`','-']
			elif choice is 't':
				obj = Character(0,level.pos,2,"./designs/tunnel.txt")
				obj.cannotCross=['T','|','/','\\','`','-']
			elif choice is 'p':
				obj = Character(0,level.pos,2,"./designs/pit.txt")
				obj.cannotCross=['/','\\','`','-']
			elif choice is 'm':
				obj = Character(0,level.pos,2,"./designs/coinbrick.txt")
				obj.cannotCross=['T','/','\\','`','-']
			elif choice is 'h':
				obj = Character(0,level.pos,2,"./designs/life.txt")
				obj.cannotCross=['T','/','\\','`','-']
			elif choice is 'g':
				obj = Character(0,level.pos,2,"./designs/gun.txt")
				obj.cannotCross=['T','|','/','\\','`','-']
			elif choice is 'e':
				obj = Character(1,level.pos,2,"./designs/goomba1.txt","./designs/goomba2.txt")
				obj.cannotCross=['T','|','/','\\','`']
			elif choice is 'q':
				break
			else:
				continue

			rules.design = ["| w : up | s : down | a : left | d : right | x : delete | p : place |"]
			while True:
				os.system("clear")
				level.printOnTop(obj,*enemies,rules)

				move = getchS()

				if move is 'a':
					if obj.x > level.pos:
						obj.moveLeft(level)
	
				elif move is 'd':
					if obj.x < 250:
						if obj.x < level.pos+(40)-obj.width:
							obj.moveRight(level)
						else:
							if obj.moveRight(level):
								level.pos+=1
								rules.x+=1
	
				elif move is 'w':
					if obj.y>0:
						obj.moveUp(level)

				elif move is 's':
					if obj.y < 23-obj.height:
						obj.moveDown(level)

				elif move is 'x':
					break

				elif move is 'p':
					if choice is 'e':
						enemies.append(obj)
						enemystr = "Goomba(1,"+str(obj.x)+","+str(obj.y)+",\"./designs/goomba1.txt\",\"./designs/goomba2.txt\")"
						enemiesstr.append(enemystr)
					else:
						level.place(obj)
					break

		if ch is not '1':
			print("Enter Level Name : ",end='')
			filename = input()

		open("./levels/"+filename+".txt", "w").close()
		file = open("./levels/"+filename+".txt","a")

		open("./levels/"+filename+"Enemy.txt", "w").close()
		enemyfile = open("./levels/"+filename+"Enemy.txt","a")

		for line in level.design:
			file.write(line+"\n")

		for line in enemiesstr:
			enemyfile.write(line+"\n")

	main()