Python User.create_row示例

示例#1

    def test_create_row(self, Boto3Resource):
        #Boto3Resource.Table.put_item.return_value = True

        test_user = User()

        test_user.table.put_item.side_effect = lambda *a, **kw: 'MOCKED'

        now = datetime.now()
        u = test_user.create_row(phonenumber='XXX',
                                 fmt_time=now.strftime('%Y-%m-%d %H:%M'))

        assert u == 'MOCKED'
        test_user.table.put_item.assert_called_once()

        u = test_user.create_row()
        assert u is None

示例#2

文件： bot_helpers.py 项目： mottaquikarim/strongbot

def find_user(params):
    user_ph = params.get('From', None)

    if not user_ph:
        return UserStatus.NO_NUMBER, None

    u = User()
    user_ph = urllib.parse.unquote(user_ph)

    user_logs = u.search_by_number(
        phonenumber=user_ph,
    )

    if not user_logs:
        u.create_row(
            phonenumber=user_ph, 
        )

        return UserStatus.NEW_USER, u


    return UserStatus.EXISTING_USER, u