db.py 文件源码

def post_add_group_member_sql(gid, uid):
    q = DBS.query(User).filter_by(uid=uid)
    gid_old = q.first().gid
    q.update({'gid': gid})

    q = DBS.query(User).filter_by(gid=gid_old)
    r = q.first()

    if not r:
        q = DBS.query(Group).filter_by(gid=gid_old)
        q.delete()
        q = DBS.query(Role_assigned).filter_by(gid=gid_old)
        q.delete()

        pid_old = DBS.query(func.min(Project.pid)).filter_by(gid=gid_old) \
            .first()[0]
        while pid_old:
            r = DBS.query(func.max(Project.pid)).filter_by(gid=gid).first()[0]
            pid = 1 if not r else r + 1

            q = DBS.query(Project).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Task).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Assign).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Comment).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Workflow).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Stage).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            q = DBS.query(Log).filter_by(gid=gid_old, pid=pid_old)
            q.update({'gid': gid, 'pid': pid})

            pid_old = DBS.query(func.min(Project.pid)).filter_by(gid=gid_old) \
                .first()[0]

    transaction.commit()


# remove member from group and add them to their own group