Python database.DBSession类代码示例

def upload_log(request):
    """
    Store new piece of log sent by tinyputer. Logs can be sent as a response to 'upload_log' command
    which created placeholder LogRequest objects.

    Parameters:
        log_request_id
        log_request_hash
    """
    try:
        log_request_id = int(request.matchdict['log_request_id'])
        log_request_hash = str(request.matchdict['log_request_hash'])
    except (KeyError, ValueError):
        raise httpexceptions.HTTPNotFound("Invalid request arguments.")
    try:
        log_request = DBSession.query(LogRequest).filter(
            LogRequest.id == log_request_id,
            LogRequest.hash == log_request_hash,
            LogRequest.log == None
        ).one()
    except database_exceptions.NoResultFound:
        raise httpexceptions.HTTPNotFound("Log request not found.")
    log_request.log = request.body
    DBSession.add(log_request)
    return Response()

def activate(request):
    successful_activation = False
    try:
        user = DBSession.query(User).filter(
            User.email == request.matchdict['email'],
            User.email_activation_code ==
                request.matchdict['email_activation_code'],
            User.activated == None
        ).one()
    except database_exceptions.NoResultFound:
        raise httpexceptions.HTTPNotFound()
    activation_form = forms.ActivationForm(csrf_context=request)
    if request.method == 'POST':
        activation_form.process(request.POST)
        if activation_form.validate():
            if (user.phone_activation_code.upper() !=
                    activation_form.phone_activation_code.data.upper()):
                request.session.flash(
                    _("The activation code you entered is invalid."),
                    queue='error'
                )
            else:
                # Regenerate activation code so user cannot reactivate his
                # account after being banned.
                user.regenerate_email_activation_code()
                user.activated = pytz.utc.localize(datetime.datetime.utcnow())
                DBSession.add(user)
                successful_activation = True
    return {
        'activation_form': activation_form,
        'successful_activation': successful_activation,
    }

 def setUp(self):
     super(TestActivate, self).setUp()
     feed_template = FeedTemplate(
         id=1,
         name='Feed template 1',
         modbus_bandwidth=9600,
         modbus_port='/dev/ttyS0',
         modbus_parity='EVEN',
         modbus_data_bits=8,
         modbus_stop_bits=1,
         modbus_timeout=5,
         modbus_endianness='BIG',
         modbus_polling_interval=120,
         dashboard_layout='IMAGE',
     )
     feed = Feed(
         id=1,
         template=feed_template,
         name=u"Feed 1",
         description=u"Description",
         latitude=20.5,
         longitude=15.3,
         created=pytz.utc.localize(datetime.datetime.utcnow()),
         api_key='some-api-key',
         activation_code='some-activation-code',
         activation_code_regenerated=pytz.utc.localize(
             datetime.datetime.utcnow()
         ),
     )
     self.activation_code = feed.activation_code
     DBSession.add_all([
         feed_template,
         feed,
     ])

 def test_normal_operation(self):
     request = self.get_request(1)
     response = views.commands(request)
     self.assertEqual(response.status_code, 200)
     self.assertDictEqual(response.json_body, {
         'commands': [],
     })
     command = Command(
         feed_id=1,
         date=_utc_datetime(2012, 5, 17, 23, 4, 11),
         command='upload_log',
         arguments={
             'url': 'http://example.org',
         }
     )
     #with transaction.manager:
     DBSession.add(command)
     request = self.get_request(1)
     response = views.commands(request)
     self.assertEqual(response.status_code, 200)
     self.assertDictEqual(response.json_body, {
         u'commands': [
             {
                 u'command': u'upload_log',
                 u'arguments': {
                     u'url': u'http://example.org',
                 },
             },
         ],
     })

 def __call__(self):
     update_chart_form = forms.UpdateChartDefinitionForm(
         self.chart_definition.id,
         self.feed,
         self.feed.template.data_stream_templates,
         csrf_context=self.request
     )
     update_chart_form.process(self.request.POST)
     data_stream_templates_dict = dict((
         (template.id, template)
         for template in self.feed.template.data_stream_templates
     ))
     if update_chart_form.validate():
         update_chart_form.populate_obj(self.chart_definition)
         self.chart_definition.data_stream_templates = []
         for id in update_chart_form.data_stream_template_ids.data:
             self.chart_definition.data_stream_templates.append(
                 data_stream_templates_dict[id]
             )
         DBSession.add(self.chart_definition)
         return Response(
             json.dumps({
                 'redirect': None,
             }),
             content_type='application/json'
         )
     return Response(
         json.dumps({
             'errors': update_chart_form.errors,
         }),
         content_type='application/json'
     )

 def test_reset_requested_value(self):
     # Request value in data stream.
     self.assertEqual(
         DBSession.query(DataStream).filter(
             DataStream.id == 2
         ).value('requested_value'),
         1234
     )
     # Pretend value has not yet been set by tinyputer - requested_value
     # still exists.
     request = self.get_request(1, 'data_stream_2', json_body={
         'datapoints': [
             {
                 'at': '2012-10-13T17:01:00.345123Z',
                 'value': '134.2344',
             },
             {
                 'at': '2012-10-13T17:02:00.425Z',
                 'value': '-23.24525',
             },
         ]
     })
     response = views.feed_data_stream(request)
     self.assertEqual(response.status_code, 200)
     self.assertEqual(
         DBSession.query(DataStream).filter(
             DataStream.id == 2
         ).value('requested_value'),
         1234
     )
     # Pretend value has been set by tinyputer - requested_value reset.
     request = self.get_request(1, 'data_stream_2', json_body={
         'datapoints': [
             {
                 'at': '2012-10-13T17:02:30.345123Z',
                 'value': '1234',
             },
         ]
     })
     #with transaction.manager
     response = views.feed_data_stream(request)
     self.assertEqual(response.status_code, 200)
     self.assertEqual(
         DBSession.query(DataStream).filter(
             DataStream.id == 2
         ).value('requested_value'),
         None
     )
     request = self.get_request(1, 'data_stream_2', json_body={
         'datapoints': [
             {
                 'at': '2012-10-13T17:03:30.345123Z',
                 'value': '4321',
             },
         ]
     })
     response = views.feed_data_stream(request)
     self.assertEqual(response.status_code, 200)

def main(argv=sys.argv):
    if len(argv) != 2:
        usage(argv)
    config_uri = argv[1]
    setup_logging(config_uri)
    settings = get_appsettings(config_uri)
    database_engine = sqlalchemy.engine_from_config(settings, 'database.')
    DBSession.configure(bind=database_engine)
    generate_data()

def insert_data_points(feed_id, data_stream, data_points, message_service):
    Measurements().insert_data_points(data_stream.id, data_points)
    # FIXME This may be not wise for production use due to race condition
    # concerns.
    dates = map(lambda x: x[0], data_points)
    HourlyAggregates().recalculate_aggregates(data_stream.id, dates)
    DailyAggregates().recalculate_aggregates(data_stream.id, dates)
    MonthlyAggregates().recalculate_aggregates(data_stream.id, dates)
    LastDataPoints().update(feed_id, data_stream.id)
    last_date, last_value = LastDataPoints().get_last_data_stream_data_point(
        feed_id,
        data_stream.id
    )
    last_value = decimal.Decimal(last_value)
    # Turn alarms associated with datastreams on/off
    alarm_definitions = DBSession.query(AlarmDefinition).filter(
        AlarmDefinition.data_stream_id == data_stream.id
    )
    alarm_messages = []
    alarms_on = []
    alarms_off = []
    for alarm_definition in alarm_definitions:
        message_content = alarm_definition.check_value(last_value)
        if message_content is None:
            alarms_off.append(alarm_definition.id)
        else:
            alarms_on.append(alarm_definition.id)
            alarm_messages.append(
                Message(
                    feed=alarm_definition.data_stream.feed,
                    data_stream=alarm_definition.data_stream,
                    message_type='ERROR',
                    date=last_date,
                    content=message_content
                )
            )
    Alarms().set_alarms_on(feed_id, data_stream.id, alarms_on, last_date)
    Alarms().set_alarms_off(feed_id, data_stream.id, alarms_off)
    for message in alarm_messages:
        message_service.create_message(message)
    # Set requested value to None if applied.
    if data_stream.writable and data_stream.requested_value is not None:
        error = abs(data_stream.requested_value - last_value)
        maximal_error = (
            decimal.Decimal(str(sys.float_info.epsilon)) * max((
                2 ** -1022,
                abs(data_stream.requested_value),
                abs(last_value))
            )
        )
        if error <= maximal_error:
            DBSession.query(DataStream).filter(
                DataStream.id == data_stream.id
            ).update({
                'requested_value': None,
                'value_requested_at': None,
            })

def main(argv=sys.argv):
    if len(argv) != 2:
        usage(argv)
    config_uri = argv[1]
    setup_logging(config_uri)
    settings = get_appsettings(config_uri)
    engine = engine_from_config(settings, 'database.')
    DBSession.configure(bind=engine)
    ModelBase.metadata.create_all(engine)

def feed_template_delete(request):
    try:
        feed_template = DBSession.query(FeedTemplate).filter(
            FeedTemplate.id == request.matchdict['feed_template_id']
        ).one()
    except database_exceptions.NoResultFound:
        raise httpexceptions.HTTPNotFound()
    delete_form = forms.DeleteFeedTemplateForm(
        feed_template_id=feed_template.id,
        csrf_context=request
    )
    delete_form.process(request.POST)
    if delete_form.validate():
        dependent_feed_count = DBSession.query(Feed).filter(
            Feed.template == feed_template
        ).count()
        dependent_data_stream_template_count = DBSession.query(
            DataStreamTemplate
        ).filter(
            DataStreamTemplate.feed_template == feed_template
        ).count()
        if dependent_feed_count > 0:
            request.session.flash(
                _(
                    "Device template cannot be deleted as there are"
                    " already devices based on it. Contact the system"
                    " support for further information."
                ),
                queue='error'
            )
        elif dependent_data_stream_template_count > 0:
            request.session.flash(
                _(
                    "Device template cannot be deleted as it has parameter"
                    " templates assigned to it. Contact the system support for"
                    " further information."
                ),
                queue='error'
            )
        else:
            DBSession.delete(feed_template)
            request.session.flash(
                _("Device template has been successfully deleted."),
                queue='success'
            )
    else:
        request.session.flash(
            _(
                "There were some problems with your request."
                " Contact the system support."
            ),
            queue='error'
        )
    return httpexceptions.HTTPFound(
        request.route_url('sngconnect.devices.feed_templates')
    )

def data_stream_template_delete(request):
    try:
        feed_template, data_stream_template = DBSession.query(
            FeedTemplate,
            DataStreamTemplate
        ).filter(
            FeedTemplate.id == request.matchdict['feed_template_id'],
            (DataStreamTemplate.id ==
                request.matchdict['data_stream_template_id']),
            DataStreamTemplate.feed_template_id == FeedTemplate.id
        ).one()
    except database_exceptions.NoResultFound:
        raise httpexceptions.HTTPNotFound()
    delete_form = forms.DeleteDataStreamTemplateForm(
        data_stream_template_id=data_stream_template.id,
        csrf_context=request,
        locale=get_locale_name(request)
    )
    delete_form.process(request.POST)
    if delete_form.validate():
        dependent_count = DBSession.query(DataStream).filter(
            DataStream.template == data_stream_template
        ).count()
        if dependent_count == 0:
            DBSession.delete(data_stream_template)
            request.session.flash(
                _("Parameter template has been successfully deleted."),
                queue='success'
            )
        else:
            request.session.flash(
                _(
                    "Parameter template cannot be deleted as there are"
                    " already parameters based on it. Contact the system"
                    " support for further information."
                ),
                queue='error'
            )
    else:
        request.session.flash(
            _(
                "There were some problems with your request."
                " Contact the system support."
            ),
            queue='error'
        )
    return httpexceptions.HTTPFound(
        request.route_url(
            'sngconnect.devices.feed_template',
            feed_template_id=feed_template.id
        )
    )

def sing_up(request):
    sign_up_form = forms.SignUpForm(csrf_context=request)
    successful_submission = False
    if request.method == 'POST':
        sign_up_form.process(request.POST)
        if sign_up_form.validate():
            user = User(
                email=sign_up_form.email.data.lower(),
                phone=sign_up_form.phone.data,
                name=sign_up_form.user_name.data,
                company_name=(
                    sign_up_form.company_name.data
                    if sign_up_form.company_name.data else
                    None
                ),
                timezone_tzname=str(request.registry['default_timezone'])
            )
            user.set_password(sign_up_form.password.data)
            DBSession.add(user)
            template = request.registry['jinja2_environment'].get_template(
                'sngconnect.accounts:templates/emails/account_activation.txt'
            )
            activation_email = EmailMessage(
                subject=_("Activate your account at SNG Connect"),
                sender=request.registry['settings']['mail.sender'],
                recipients=[user.email],
                body=template.render(
                    activation_url=request.route_url(
                        'sngconnect.accounts.activate',
                        email=user.email,
                        email_activation_code=user.email_activation_code
                    )
                )
            )
            get_mailer(request).send(activation_email)
            sms_service = SMSService(request.registry)
            sms_service.send_sms(
                [user.phone],
                _(
                    "Your SNG:connect confirmation code is: ${code}",
                    mapping={
                        'code': user.phone_activation_code,
                    }
                )
            )
            successful_submission = True
    return {
        'sign_up_form': sign_up_form,
        'successful_submission': successful_submission,
    }

 def set_requested_value(self, data_stream, value):
     DBSession.query(DataStream).filter(
         DataStream.id == data_stream.id
     ).update({
         'requested_value': value,
         'value_requested_at': pytz.utc.localize(
             datetime.datetime.utcnow()
         ),
     })
     self.registry['scheduler'].add_date_job(
         DataStreamService.assert_requested_value,
         datetime.datetime.now() + self._requested_value_timeout,
         [self.registry, data_stream.id]
     )

 def validate_name(self, field):
     if field.errors:
         return
     try:
         DBSession.query(ChartDefinition).filter(
             ChartDefinition.feed_template == self.feed_template,
             ChartDefinition.feed == None,
             ChartDefinition.name == field.data
         ).one()
     except database_exceptions.NoResultFound:
         pass
     else:
         raise validators.ValidationError(
             _("This chart name is already taken.")
         )

def generate_data():
    user = User(
        name='User',
        email='[email protected]',
        phone='+48123456789',
        activated=pytz.utc.localize(datetime.datetime.utcnow()),
        role_user=True,
        timezone_tzname='Europe/Warsaw'
    )
    user.set_password('user')
    kid = User(
        name='Kid',
        email='[email protected]',
        phone='+48123456789',
        activated=pytz.utc.localize(datetime.datetime.utcnow()),
        role_user=True,
        timezone_tzname='Europe/Warsaw'
    )
    kid.set_password('kid')
    maintainer = User(
        name='Maintainer',
        email='[email protected]',
        phone='+48123456789',
        activated=pytz.utc.localize(datetime.datetime.utcnow()),
        role_maintainer=True,
        timezone_tzname='Europe/Warsaw'
    )
    maintainer.set_password('maintainer')
    supplier = User(
        name='Supplier',
        email='[email protected]',
        phone='+48123456789',
        activated=pytz.utc.localize(datetime.datetime.utcnow()),
        role_supplier=True,
        timezone_tzname='Europe/Warsaw'
    )
    supplier.set_password('supplier')
    admin = User(
        name='Admin',
        email='[email protected]',
        phone='+48123456789',
        activated=pytz.utc.localize(datetime.datetime.utcnow()),
        role_administrator=True,
        timezone_tzname='Europe/Warsaw'
    )
    admin.set_password('admin')
    DBSession.add_all([user, kid, maintainer, supplier, admin])
    transaction.commit()