def update(new_post, categories):
    """Update post.

    Arguments:
    new_post -- Posts object
    categories - list of categories
    """

    session = DBSession()

    # get data
    post = session.query(Posts).filter(Posts.id == new_post.id,
                                       Posts.is_deleted == False).one()

    # update data
    if new_post.header is not None:
        post.header = new_post.header

    if new_post.content is not None:
        post.content = new_post.content

    if categories:
        post.categories.clear()
        for c in categories:
            category = session.query(Categories).get(c)
            post.categories.append(category)

    post.modification_date = datetime.now().replace(
        microsecond=0).isoformat(' ')

    session.commit()
    session.close()
def get_all():
    """Return list of all categories"""
    session = DBSession()

    result = session.query(Categories).all()

    session.close()
    return result
def get(id):
    """Return category by id"""
    session = DBSession()

    category = session.query(Categories).get(id)

    session.close()
    return category
def add(category):
    """Add new category"""
    session = DBSession()

    session.add(category)

    session.commit()
    session.close()
Example #5
0
def add(user):
    """Add new user"""
    session = DBSession()

    session.add(user)

    session.commit()
    session.close()
Example #6
0
def get(id):
    """Get user by id"""
    session = DBSession()

    user = session.query(Users).get(id)

    session.close()
    return user
def get_author(id):
    """Return author id by post's id."""
    session = DBSession()

    post = session.query(Posts).filter(Posts.id == id). \
        options(load_only('author_id')).one()

    session.close()
    return post.author_id
def get_all():
    """Return list of all posts."""
    session = DBSession()

    posts = session.query(Posts).filter(Posts.is_deleted == False). \
        order_by(Posts.id.desc()).all()

    session.close()
    return posts
def get(id):
    """Return post by it's id."""
    session = DBSession()

    post = session.query(Posts).filter(Posts.id == id,
                                       Posts.is_deleted == False).one()

    session.close()
    return post
def get_all_by_category(id):
    """Return list of all posts in the category by it's id."""
    session = DBSession()

    posts = session.query(Posts).join(Posts.categories). \
        filter(Categories.id == id, Posts.is_deleted == False). \
        order_by(Posts.id.desc()).all()

    session.close()
    return posts
def delete(id):
    """Delete post by id"""
    session = DBSession()

    post = session.query(Posts).filter(Posts.id == id,
                                       Posts.is_deleted == False).one()

    # actually just set is_deleted flag to True
    post.is_deleted = True
    post.deletion_date = datetime.now().replace(microsecond=0).isoformat(' ')

    session.commit()
    session.close()
def restore(post_id):
    """Restore deleted post by id"""
    session = DBSession()

    post = session.query(Posts).filter(Posts.id == post_id,
                                       Posts.is_deleted == True).one()

    post.modification_date = datetime.now().replace(
        microsecond=0).isoformat(' ')
    post.deletion_date = None
    post.is_deleted = False

    session.commit()
    session.close()
Example #13
0
def validate(username, password):
    """Validate user.
    If successfull - returns user's id, else - None.
    """
    session = DBSession()

    user = session.query(Users).filter(Users.username == username).first()

    session.close()

    if user is not None and user.password == password:
        return user.id
    else:
        return None
def get_custom_by_category(category_id, quantity, from_id=None, newer=False):
    """Return list of posts by custom rules in the selected category.

    Arguments:
    category_id -- id of the category
    quantity -- quantity of posts
    from_id -- start from this post
    newer --
        if False - returns older posts
        if True - return newer posts
    """

    session = DBSession()

    # start from last post if no from_id passed
    if from_id is None:
        posts = session.query(Posts).join(Posts.categories). \
            filter(Categories.id == category_id, Posts.is_deleted == False). \
            order_by(Posts.id.desc()). \
            limit(int(quantity)).all()

    # start from custom post
    else:

        # get older posts
        if not newer:
            posts = session.query(Posts).join(Posts.categories). \
                filter(Categories.id == category_id,
                       Posts.id <= int(from_id),
                       Posts.is_deleted == False). \
                order_by(Posts.id.desc()). \
                limit(int(quantity)). \
                all()

        # get newer posts
        else:
            posts = session.query(Posts).join(Posts.categories). \
                filter(Categories.id == category_id,
                       Posts.id >= int(from_id),
                       Posts.is_deleted == False). \
                order_by(Posts.id). \
                limit(int(quantity)). \
                from_self(). \
                order_by(Posts.id.desc()). \
                all()

    session.close()
    return posts
def get_first_last():
    """Return id of the first and the last post

    Returns dictionary, keys : first_id, last_id.
    """

    session = DBSession()

    first_id = session.query(Posts).filter(Posts.is_deleted == False). \
        order_by(Posts.id).first().id

    last_id = session.query(Posts).filter(Posts.is_deleted == False). \
        order_by(Posts.id.desc()).first().id

    session.close()
    return {'first_id': first_id, 'last_id': last_id}
def add(post, categories):
    """Add new post and return it's id.

    Arguments:
    post - Posts object
    categories - list of categories id
    """

    session = DBSession()

    session.add(post)
    # add categories to the post
    for id in categories:
        category = session.query(Categories).get(id)
        post.categories.append(category)

    id = post.id

    session.commit()
    session.close()
    return id
Example #17
0
        return a_all_list

    def op_del(self):
        pass

    def op_chg(self):
        pass


if __name__ == '__main__':
    engine = create_engine(
        'mysql+pymysql://cdn_user:[email protected]:3306/cdn?charset=utf8'
    )
    Base.metadata.create_all(engine)
    DBSession = sessionmaker(bind=engine)
    session = DBSession()

    op = operation(session)

    data_dic = {
        'cname': [['www.baidu.com', 'a.shifen.com', 0, '8.8.8.8'],
                  ['www.sina.com', 'w.sina.com', 0, '114.114.8.8'],
                  ['www.tent.com', 'tt.asdf.com', 0, '110.0.110.8'],
                  ['www.ali.com', 'ali.ali.com', 0, '123.123.123.8']],
        'a': [['a.shifen.com', '192.168.2.3', '8.8.8.8', '192.168.2.3', 2],
              ['w.sina.com', '192.168.6.16', '114.114.8.8', '192.168.2.3', 2],
              ['tt.asdf.com', '192.127.8.3', '110.0.110.8', '192.168.2.3', 2],
              ['ali.ali.com', '192.26.2.3', '123.123.123.8', '192.168.2.3', 2]]
    }

    # op.op_add(data_dic)