Exemplo n.º 1
0
def userprofile_bind_group(argv):
    if "-f" in argv:
        file = argv[argv.index("-f") + 1]
        filename = "%s/%s/%s" % (Base, "conf", file)
        f = open(filename, "r")
        info = yaml.load(f)
        for key, val in info.items():
            print(key)
            print(val)
            from modules.data_init import Userprofiles, Groups

            userprofile_2_group = session.query(Userprofiles).filter(Userprofiles.user == key).first()
            group = session.query(Groups).filter(Groups.groupname == val).first()
            userprofile_2_group.group = [group]
            # session.add(g1)
            session.commit()
    else:
        print(
            """
        -f 指定配置文件
        """
        )
Exemplo n.º 2
0
def userprofile_bind_user(argv):
    if "-f" in argv:
        file = argv[argv.index("-f") + 1]
        filename = "%s/%s/%s" % (Base, "conf", file)
        f = open(filename, "r")
        info = yaml.load(f)
        for key, val in info.items():
            print(key)
            print(val)
            from modules.data_init import Userprofiles, Hostusers

            userprofile_2_user = session.query(Userprofiles).filter(Userprofiles.user == key).first()
            for i in val:
                user = session.query(Hostusers).filter(Hostusers.user == i).first()
                print(user)
                userprofile_2_user.hostuser = [user]
                # session.add(g1)
                session.commit()
    else:
        print(
            """
        -f 指定配置文件
        """
        )