Python Wrapper.select方法代码示例

# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import select [as 别名]
class Statuses(object):
	"""docstring for Statuses"""

	def __init__(self):
		self.wrapper = Wrapper()
	def get_all(self):
		return self.wrapper.select(["id", "status"], "statuses")
		

# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import select [as 别名]
class Cooker(object):
    """docstring for Cooker"""


    def __init__(self):
        self.wrapper = Wrapper()



    def advanced_search(self, diction):
        return advanced_search(
            ["distinct dishes.id", "name", "description", "price",
             "status", "image", "id_status", "count", "id_category"],
             "dishes inner join statuses on dishes.id_status=statuses.id",
            diction)


    def search_dishes(self, string):
        return self.wrapper.select(["distinct dishes.id", "name", "description", "price", "status",
                                    "image", "id_status", "count", "id_category"],
                                    "dishes",
                                    "inner join statuses on statuses.id=id_status "\
                                    "where lower(name) LIKE lower('%{0}%') or "\
                                    "lower(description) LIKE lower('%{0}%') or "\
                                    "lower(price) LIKE lower('%{0}%') or "\
                                    "lower(status) LIKE lower('%{0}%') or "\
                                    "lower(count) LIKE lower('%{0}%')".format(string))


    def delete_item_menu(self, id_item_menu):
        return self.wrapper.update({"id_status":2},
                                    "dishes",
                                    "where id={0}".format(id_item_menu))

    def get_all_categories(self):
        return self.wrapper.select(["category", "id"], "categories")


    def get_categories_from_inner_dishes(self):
    	dictionary = {}
    	for category in self.wrapper.select(["category", "id"], "categories"):
    		dishes = self.wrapper.select("*", "dishes", "where id_category=%s" % (category['id']))
    		dictionary[category['category']] = list(dishes)
    	return dictionary


    def get_menu_by_category(self, id_category):
        return self.wrapper.select(["dishes.id", "name", "description", "price", "status",
                                    "image", "id_status", "count", "id_category"],
                                    ["dishes", "statuses"],
                                    "where id_category=%s and id_status=statuses.id" % (id_category))


    def get_item_menu(self, id_item_menu):
        return self.wrapper.select(["dishes.id", "name", "description", "price", "category",
                                    "image", "id_status", "count", "id_category"],
                                   ["dishes", "categories"],
                                   "where dishes.id={0} and id_category=categories.id".format(id_item_menu))


    def get_all_dishes(self):
        """ Get all dishes from DB """
        return self.wrapper.select(["dishes.id", "name", "description", "price", 
                                    "image", "count", "status", "id_status", "id_category"],
                                   ["dishes", "statuses"], 
                                    "where id_status=statuses.id")

    def get_dishes_by_cat(self,catid):
        """ Get dishes by categories """
        return self.wrapper.select(["*"], "dishes","where id_category={0}".format(catid))

    def add_item_category(self, item_category):
        return self.wrapper.insert(item_category, "categories")


    def add_item_menu(self, item_menu):
        return self.wrapper.insert(item_menu, "dishes")


    def edit_item_category(self, id_category, item_category):
        return self.wrapper.update(item_category, "categories", "where id={0}".format(id_category))


    def edit_item_menu(self, id_menu, item_menu):
        return self.wrapper.update(item_menu, "dishes", "where id={0}".format(id_menu))

# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import select [as 别名]
class Users(object):

    """ Users Class
                    Methods:
                            get_user()
                            get_all_users()
                            get_permission()
                            login()
                            validateuser()
                            is_pwd()
                            is_email()
                            is_username()
                            is_number()
    """
    db_name = "users"

    def __init__(self):
        self.w = Wrapper()

    def __del__(self):
        pass

    def get_user(self, uid):
        """ Get User
            Return all user row.
        """
        return self.w.select(["*"], [self.db_name], "where id={0}".format(uid))

    def get_all_users(self):
        """ Get All user method
            Return list of users
        """
        return self.w.select(["*"], [self.db_name])

    def get_permission(self, uname):
        """ Get user permission method
            Return id users role
        """
        return self.w.select("id_role", self.db_name, "where login='{0}'".format(uname))

    def validateuser(self, credendials):
        """ Validate all users fields
        """
        if isinstance(credendials, dict):
            if self.is_username(credendials['username']) \
                and self.is_pwd(credendials['password']) \
                and self.is_email(credendials['mail']) \
                and self.is_username(credendials['f_name']) \
                and self.is_username(credendials['l_name']) \
                and self.is_number(credendials['status']) \
                and self.is_number(credendials['id_role']):
                return True
            else:
                return False
        else:
            return {"error": "need dict"}

    def login(self, credendials):
        """ Login user """
        try:
            if isinstance(credendials, dict):
                dbuser = self.w.select(
                    "login", self.db_name, "where login='{0}'".format(credendials['username']))
                if dbuser[0]['login'] == format(credendials['username']):
                    dbpass = self.w.select(
                        "password", self.db_name, "where login='{0}'".format(credendials['username']))
                    if dbpass[0]['password'] == credendials['password']:
                        return True
                    else:
                        return False
                else:
                    return False
            else:
                return {"error": "need dict"}
        except Exception as e:
            return False

    def search_users(self, param):
        """ search """
        search_str = param.split(' ')
        if(len(search_str) == 1):
            return self.w.select("id,f_name,l_name,login,email", self.db_name, "where id like '%{0}%' or f_name like '%{0}%' or l_name like '%{0}%' or login like '%{0}%' or email like '%{0}%'".format(self.search_validation(param.lower())))
        else:
            xdic = []
            for x in search_str:
                xxx = self.w.select(
                    "id,f_name,l_name,login,email", self.db_name, "where id like '%{0}%' or f_name like '%{0}%' or l_name like '%{0}%' or login like '%{0}%' or email like '%{0}%'".format(self.search_validation(x.lower())))
                xdic.append(xxx[0])
            return xdic

    def advanced_search_users(self, id, f_name, l_name, login, email):
        """ Advanced search """
        return self.w.select("id,f_name,l_name,login,email", self.db_name, "where id like '%{0}%' and f_name like '%{1}%' and l_name like '%{2}%' and login like '%{3}%' and email like '%{4}%'".format(id, f_name, l_name, login, email))

    def search_validation(self, param):
        out = re.sub('[\W]', '', param)
        return out

    def user_count(self):
        return len(self.w.select('id',self.db_name))
#.........这里部分代码省略.........

# 需要导入模块: from wrapper import Wrapper [as 别名]
# 或者: from wrapper.Wrapper import select [as 别名]
class  Manager(Waiter):
    """in data we put all fields we need to see in ticket!"""

    def __init__(self):
        self.wrap = Wrapper()

    
    def get_full_order(self, order_id):
        """in data put all what we need to see in ticket"""
        data = "dishes.image, dishes.name, dishes.price, dishes.id as id_dish, tickets.count, tickets.id, tickets.id_order"
        condition = "INNER JOIN dishes ON tickets.id_dish = dishes.id WHERE tickets.id_order = %s" % (order_id)
        tickets = self.wrap.select(data, "tickets", condition)
        
        full_price = 0
        if len(tickets) != 0:
            """for ticket in tickets:
                full_price += ticket["price"]
            tickets += ({"full_price": full_price},)"""
            return tickets
        return None
    
    def get_all_orders(self, Date = ''):
        """get all orders all waiters for curent date"""
        if Date == '':
            date = datetime.datetime.now().strftime('%Y-%m-%d')
        else:
            date = Date

        data = "orders.id, statuses.status, CAST(Sum(tickets.price) as UNSIGNED) as TotalCount"
        condition = "JOIN orders on orders.id=tickets.id_order \
        WHERE orders.id_status = statuses.id \
        AND (orders.id_status = 4 OR orders.id_status =5) \
        AND orders.id_user = users.id \
        AND orders.date LIKE '{0}%' \
        GROUP BY tickets.id_order".format(date)
        
        orders = self.wrap.select(data, "statuses, users, tickets", condition)
        if len(orders) != 0:
            return orders
        return None
  
    def edit_order(self, order_data):
        """get order data with order id
        [{count": 1,"id": 15},..."""
        id = order_data["id"]
        del order_data["id"]
        del order_data["name"]
        del order_data["image"]
        del order_data["price"]
        self.wrap.update(order_data, "tickets", "WHERE tickets.id = %s" % (id))

    def del_ticket(self, ticket_id):
        """update ticket status to 0, default 1"""
        return self.wrap.delete("tickets", ticket_id)

    def get_order(self, order_id):
        tickets = self.wrap.select("*", "tickets",
                          "WHERE tickets.id_order = %s" % (order_id))
        return tickets

    def remove_order(self, order_id):
        """get order id. Set status to NULL"""
        self.wrap.update({"id_status": 3}, "orders",
                          "WHERE id = %s" % (order_id))

    def get_user_id(self, username):
        data = "users.id"
        table = "users"
        condition = "WHERE users.login = '%s'" % (username)
        resp = self.wrap.select(data, table, condition)
        return resp[0]['id']

    def get_summ(self):
        data = "orders.id, Sum(tickets.price) as TotalCount"
        condition = "JOIN orders on orders.id=tickets.id_order GROUP BY tickets.id_order"

        orders = self.wrap.select(data, "tickets", condition)
        return orders