簡體   English   中英

如何使用API​​將os.fork與sqlalchemy core和flask一起使用?

[英]How to use os.fork with sqlalchemy core and flask using API?

我已經創建了一個用戶,並希望將電子郵件發送給該用戶的郵件ID,並提供詳細信息:

def send_new_user_mail(new_user_id, plain_passwd):
    '''
    Send mail to new user added
    '''
    s1 = select([users]).where(users.c.id == new_user_id)
    rs1 = conn.execute(s1).fetchone()
    if rs1:
        msg = Message("You are registered to use DMS",
                      sender=app.config['DEFAULT_MAIL_SENDER'],
                      recipients=[rs1[users.c.email]])
        print g.referer, 'g.referer'
        user_details = ('username: ' +  rs1[users.c.email] + '<br>' +
                        'password:' + plain_passwd)
        # TODO: send change password link
        chg_passwd = ('<br><br>You can change password by visiting: ' +
                      '<a href=url_for("change_password")>' +
                      url_for("change_my_password", user_id=rs1[users.c.id]) +
                      '</a>')
        chg_passwd = ''
        msg.html = ('url: ' + '<a href=' + str(g.referer) + '>' +
                    g.referer + '</a></br><br>' + user_details +
                    chg_passwd)
        print msg.html, 'msg.html'
        try:
            with app.app_context():
                mail.send(msg)
                return True
        except Exception, e:
            print e
            return False
    conn1.close()
    return

@app.route('/sf/api/v1.0/users', methods=['POST'])
def add_user():
    '''
    Add a user 
    '''
    if 'id' in session:
        data = request.get_json(force=True)
        trans = conn.begin()
        try:
            if 'first_name' in data:
                data['first_name'] = data['first_name'].title()
            if 'last_name' in data:
                data['last_name'] = data['last_name'].title()
            ins = users.insert().values(data)
            rs = conn.execute(ins)
            mnew_user = rs.inserted_primary_key[0]
            mfolder = create_ftp_user_source_folder(mnew_user)
            trans.commit()
            for g in mgroups:
                ins2 = users_groups.insert().values(
                    user=mnew_user,
                    group=g)
                rs2 = conn1.execute(ins2)
        except Exception, e:
            print e
            trans.rollback()
            return jsonify({'message': "Invalid / duplicate details"}), 400
        # user FTP source folder exists
        if mfolder:
            # new_child_proc = 0
            new_child_proc = os.fork()
            print new_child_proc, 'new_child_proc'
            if new_child_proc == 0:
                x = send_new_user_mail(mnew_user, plain_passwd)
                os._exit(0)
            else:
               print 'child:', new_child_proc
            return jsonify({'id': mnew_user}), 201
        else:
            return jsonify(
                {'message': "Cannot create FTP source/already exists"}), 403

    return jsonify({'message': "UNAUTHORIZED"}), 401

我嘗試了此代碼,但出現錯誤:還添加了代碼的其他部分。 如果用戶已經屬於一個組,則不會在users_groups中添加它,而是繼續。 添加新用戶后,還會在文件系統中創建一個源文件夾。 正如建議的那樣,我將值存儲在變量中后進行了分叉並嘗試過,但仍然出現相同的錯誤。 有沒有更好的解決方案呢?

(psycopg2.DatabaseError)SSL錯誤:解密失敗或錯誤的記錄mac

連接池出現一些錯誤,我在這里不知道如何使用它。 請指導。 Flask v.0.11,python 2.7,sqlalchemy核心1.0.8

按照@inverio的建議更新我的答案,方法是在fork之后刪除查詢部分:

def send_new_user_mail(memail_id, plain_passwd, mnew_user):    
    '''
    Send mail to new user added to dms with url
    '''
    msg = Message("You are registered to use DMS",
                  sender=app.config['DEFAULT_MAIL_SENDER'],
                  recipients=[memail_id])
    user_details = ('username: ' + memail_id + '<br>' +
                    'password:' + plain_passwd)
    chg_passwd = ''
    msg.html = ('url: ' + '<a href=' + str(g.referer) + '>' +
                g.referer + '</a></br><br>' + user_details +
                chg_passwd)
    try:
        with app.app_context():
            mail.send(msg)
            return True
    except Exception, e:
        print e
        return False
    conn1.close()
    return

@app.route('/sf/api/v1.0/users', methods=['POST'])
def add_user():
    '''
    Add a user and select his group(s)
    '''
    if 'id' in session:
        if session['email'].split("@")[0] == 'ytms.admin' and \
                session['flag'] is True:
            mgroups = []
            mnew_user = None
            mfolder = None
            memail = None
            data = request.get_json(force=True)
            if data['password']:
               # pass
               plain_passwd = data['password']
               mypass = pwd_context.encrypt(data['password'])
               data['password'] = mypass
            if "groups" not in data:
                return jsonify({'message': "Cannot add without group(s)"}), 400
            for g in data["groups"]:
                mgroups.append(g)
            del data["groups"]
            print data, 'data'
            trans = conn.begin()
            try:
                if 'first_name' in data:
                    data['first_name'] = data['first_name'].title()
                if 'last_name' in data:
                    data['last_name'] = data['last_name'].title()
                ins = users.insert().values(data)
                rs = conn.execute(ins)
                mnew_user = rs.inserted_primary_key[0]
                mfolder = create_ftp_user_source_folder(mnew_user)
                memail = data['email']
                trans.commit()
                for g in mgroups:
                    ins2 = users_groups.insert().values(
                        user=mnew_user,
                        group=g)
                    rs2 = conn1.execute(ins2)
            except Exception, e:
                print e
                trans.rollback()
                return jsonify({'message': "Invalid / duplicate details"}), 400
            # user FTP source folder exists
            if mfolder:
                new_child_proc = os.fork()
                if new_child_proc == 0:
                    x = send_new_user_mail(memail, plain_passwd, mnew_user)
                    os._exit(0)
                else:
                   print 'child:', new_child_proc
                return jsonify({'id': mnew_user}), 201
            else:
                return jsonify(
                    {'message': "Cannot create FTP source/already exists"}), 403

    return jsonify({'message': "UNAUTHORIZED"}), 401

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM