def testFirst(self):
        first_query = self.first_query
        session.query(Event).delete()
        session.commit()
        session.flush()

        e = Event(self.et)
        e.title = 'First event'
        e.time_start = datetime(1970, 1, 1, 0, 0)
        e.time_end = datetime(1970, 1, 1, 3, 0)
        e.description = 'First description'
        e.source_type = EventSourceType.EMPTY
        session.add(e)
        session.commit()

        all_first = first_query.all()
        self.assertEquals(len(all_first), 1)
        e = all_first[0]
        self.assertEquals(e.title, 'First event')
        self.assertEquals(e.time_start, datetime(1970, 1, 1, 0, 0))
        self.assertEquals(e.time_end, datetime(1970, 1, 1, 3, 0))
        self.assertEquals(e.description, 'First description')
        self.assertEquals(e.source_type, EventSourceType.EMPTY)
        self.assertEquals(e.event_type.name, 'first')

        all_date_empty = session.query(Event).filter(
            Event.time_start == date(1971, 1, 1)).all()
        self.assertEquals(len(all_date_empty), 0)
        all_date_fine = session.query(Event).filter(
            Event.time_start == date(1970, 1, 1)).all()
        self.assertEquals(len(all_date_fine), 1)
        session.delete(e)
        session.commit()
    def testFirst(self):
        first_query = self.first_query
        session.query(Event).delete()
        session.commit()
        session.flush()

        e = Event(self.et)
        e.title = 'First event'
        e.time_start = datetime(1970, 1, 1, 0, 0)
        e.time_end = datetime(1970, 1, 1, 3, 0)
        e.description = 'First description'
        e.source_type = EventSourceType.EMPTY 
        session.add(e)
        session.commit()

        all_first = first_query.all()
        self.assertEquals(len(all_first), 1)
        e = all_first[0]
        self.assertEquals(e.title, 'First event');
        self.assertEquals(e.time_start, datetime(1970, 1, 1, 0, 0))
        self.assertEquals(e.time_end, datetime(1970, 1, 1, 3, 0))
        self.assertEquals(e.description, 'First description')
        self.assertEquals(e.source_type, EventSourceType.EMPTY)
        self.assertEquals(e.event_type.name, 'first')

        all_date_empty = session.query(Event).filter(Event.time_start == date(1971, 1, 1)).all() 
        self.assertEquals(len(all_date_empty), 0)
        all_date_fine = session.query(Event).filter(Event.time_start == date(1970, 1, 1)).all() 
        self.assertEquals(len(all_date_fine), 1)
        session.delete(e)
        session.commit()
    def testThird(self):
        e = Event(self.et)
        e.title = 'First event'
        e.source_type = EventSourceType.EMPTY
        session.add(e)
        session.commit()

        e.event_status_list.append(EventStatus(EventStatus.EMPTY))
        e.event_status_list.append(EventStatus(EventStatus.LIVE_BE_HERE))
        self.assertEquals(e.last_status, EventStatus.LIVE_BE_HERE)

        session.delete(e)
        session.commit()
    def testThird(self):
        e = Event(self.et)
        e.title = 'First event'
        e.source_type = EventSourceType.EMPTY 
        session.add(e)
        session.commit()

        e.event_status_list.append(EventStatus(EventStatus.EMPTY))
        e.event_status_list.append(EventStatus(EventStatus.LIVE_BE_HERE))
        self.assertEquals(e.last_status, EventStatus.LIVE_BE_HERE)

        session.delete(e) 
        session.commit()
    def testDefThumb(self):
        it = ImageType(ImageType.TARGET_NONE)
        it.max_thumb_height = 345
        it.max_thumb_width = 234
        session.add(it)
        session.commit()
        ret = it.mkDefThumb()
        self.assertEquals(ret, True)
        self.assert_(it.def_thumb_path)
        fp = FileProcess()
        self.assertNotEquals(it.def_thumb_path[0], '/')
        info = ImageInfo(fp.fullPath(it.def_thumb_path))
        self.assertEquals(info.is_image(), True)
        self.assertEquals(info.height, 345)
        self.assertEquals(info.width, 234)

        session.delete(it)
        session.commit()
    def testDefThumb(self):
        it = ImageType(ImageType.TARGET_NONE)
        it.max_thumb_height = 345
        it.max_thumb_width = 234
        session.add(it)
        session.commit()
        ret = it.mkDefThumb()
        self.assertEquals(ret, True)
        self.assert_(it.def_thumb_path)
        fp = FileProcess()
        self.assertNotEquals(it.def_thumb_path[0], '/')
        info = ImageInfo(fp.fullPath(it.def_thumb_path))
        self.assertEquals(info.is_image(), True)
        self.assertEquals(info.height, 345)
        self.assertEquals(info.width, 234)

        session.delete(it)
        session.commit()
    def testAddImageToDb(self):
        it = ImageType(ImageType.TARGET_NONE)
        it.max_thumb_height = 50
        it.max_thumb_width = 150
        it.base_dir = 'b/a'
        it.transform_type = ImageTransform.STD
        session.add(it)
        img = Image(it)
        ret = img.uploadFromFile(fileInTestDir('img/test.jpg'))
        self.assert_(ret)
        session.add(img)
        session.commit()
        old_thumb_path = img.thumb_path
        old_image_path = img.image_path
        self.assert_(os.path.isfile(FileProcess.fullPath(img.thumb_path)))
        self.assert_(os.path.isfile(FileProcess.fullPath(img.image_path)))
        split_p = img.thumb_path.split(os.path.sep)
        self.assert_('b' in split_p)
        self.assert_('a' in split_p)
        self.assert_('img' in split_p)
        split_p = img.image_path.split(os.path.sep)
        self.assert_('b' in split_p)
        self.assert_('a' in split_p)
        self.assert_('img' in split_p)
        self.assert_(img.thumb_width <= 150)
        self.assert_(img.thumb_height <= 50)
        self.assertEquals(img.content_type, ImageInfo.JPEG)
        self.assertEquals(img.image_width, 418)
        self.assertEquals(img.image_height, 604)

        ret = img.uploadFromFile(fileInTestDir('img/test.jpg'))
        self.assert_(ret)
        self.assert_(not os.path.isfile(FileProcess.fullPath(old_thumb_path)))
        self.assert_(not os.path.isfile(FileProcess.fullPath(old_image_path)))

        session.delete(img)
        session.delete(it)
        session.commit()
    def testAddImageToDb(self):
        it = ImageType(ImageType.TARGET_NONE)
        it.max_thumb_height = 50
        it.max_thumb_width = 150
        it.base_dir = 'b/a'
        it.transform_type = ImageTransform.STD
        session.add(it)
        img = Image(it)
        ret = img.uploadFromFile(fileInTestDir('img/test.jpg'))
        self.assert_(ret)
        session.add(img)
        session.commit()
        old_thumb_path = img.thumb_path
        old_image_path = img.image_path
        self.assert_(os.path.isfile(FileProcess.fullPath(img.thumb_path)))
        self.assert_(os.path.isfile(FileProcess.fullPath(img.image_path)))
        split_p = img.thumb_path.split(os.path.sep)
        self.assert_('b' in split_p)
        self.assert_('a' in split_p)
        self.assert_('img' in split_p)
        split_p = img.image_path.split(os.path.sep)
        self.assert_('b' in split_p)
        self.assert_('a' in split_p)
        self.assert_('img' in split_p)
        self.assert_(img.thumb_width <= 150)
        self.assert_(img.thumb_height <= 50) 
        self.assertEquals(img.content_type, ImageInfo.JPEG)
        self.assertEquals(img.image_width, 418)
        self.assertEquals(img.image_height, 604) 

        ret = img.uploadFromFile(fileInTestDir('img/test.jpg'))
        self.assert_(ret)
        self.assert_(not os.path.isfile(FileProcess.fullPath(old_thumb_path)))
        self.assert_(not os.path.isfile(FileProcess.fullPath(old_image_path)))

        session.delete(img)
        session.delete(it)
        session.commit()
Example #9
0
	def do_POST(self):
		try:
			

			output = header
			
			ctype, pdict = cgi.parse_header(self.headers.getheader("content-type"))
			if ctype == 'multipart/form-data':
				fields = cgi.parse_multipart(self.rfile, pdict)

				restaurant = fields.get("new_rest")
				edit = fields.get("edit")

			if restaurant:
				self.send_response(301)
				self.end_headers()
				restaurant_new = Restaurant(name = restaurant[0])
				session.add(restaurant_new)
				session.commit()

				text = """
						<h2>New Restaurant was created!</h2>
						<p>Restaurant <strong>%s</strong> is added to restaurants</p>
						<a href = "/restaurants/new"><p>Back</p></a>
						<a href = "/restaurants"><p>To All restaurants</p></a>
 					""" % restaurant[0]

 				output += text
 				output += footer
				self.wfile.write(output)

 			elif edit:
 				self.send_response(301)
				self.end_headers()
 				rest_id = int(self.path.split("/")[-2])
 				new_name = edit[0]
 				old_rest = session.query(Restaurant).filter(Restaurant.id == rest_id)
 				old_name = old_rest[0].name
 				old_rest[0].name = new_name
 				session.add(old_rest[0])
 				session.commit()

 				text = """
 					<h2>You have changed the name of %s to %s.</h2>
 					<h3>Congrats!</h3>
					<a href = "/restaurants"><p>To All restaurants</p></a>
 				""" % (old_name, new_name)
 				output += text
 				output += footer
				self.wfile.write(output)

 			elif "delete" in fields.keys()[0]:
				del_command = fields.keys()[0]
				print "del del_command: ", del_command
				del_id = int(del_command.split("_")[1])
				delete_restaurant = session.query(Restaurant).filter(Restaurant.id == del_id).one()
				session.delete(delete_restaurant)
				session.commit()
				self.send_response(301)
				self.send_header('Content-type', 'text/html')
				self.send_header('Location', '/restaurants')
				self.end_headers()

		
		except Exception, e:
			print "ERROR! ", e
			pass
Example #10
0
    def do_POST(self):
        try:

            output = header

            ctype, pdict = cgi.parse_header(
                self.headers.getheader("content-type"))
            if ctype == 'multipart/form-data':
                fields = cgi.parse_multipart(self.rfile, pdict)

                restaurant = fields.get("new_rest")
                edit = fields.get("edit")

            if restaurant:
                self.send_response(301)
                self.end_headers()
                restaurant_new = Restaurant(name=restaurant[0])
                session.add(restaurant_new)
                session.commit()

                text = """
						<h2>New Restaurant was created!</h2>
						<p>Restaurant <strong>%s</strong> is added to restaurants</p>
						<a href = "/restaurants/new"><p>Back</p></a>
						<a href = "/restaurants"><p>To All restaurants</p></a>
 					""" % restaurant[0]

                output += text
                output += footer
                self.wfile.write(output)

            elif edit:
                self.send_response(301)
                self.end_headers()
                rest_id = int(self.path.split("/")[-2])
                new_name = edit[0]
                old_rest = session.query(Restaurant).filter(
                    Restaurant.id == rest_id)
                old_name = old_rest[0].name
                old_rest[0].name = new_name
                session.add(old_rest[0])
                session.commit()

                text = """
 					<h2>You have changed the name of %s to %s.</h2>
 					<h3>Congrats!</h3>
					<a href = "/restaurants"><p>To All restaurants</p></a>
 				""" % (old_name, new_name)
                output += text
                output += footer
                self.wfile.write(output)

            elif "delete" in fields.keys()[0]:
                del_command = fields.keys()[0]
                print "del del_command: ", del_command
                del_id = int(del_command.split("_")[1])
                delete_restaurant = session.query(Restaurant).filter(
                    Restaurant.id == del_id).one()
                session.delete(delete_restaurant)
                session.commit()
                self.send_response(301)
                self.send_header('Content-type', 'text/html')
                self.send_header('Location', '/restaurants')
                self.end_headers()

        except Exception, e:
            print "ERROR! ", e
            pass
 def tearDown(self):
     session.delete(self.et)
     session.commit()
 def tearDown(self):
     session.delete(self.et)
     session.commit()