################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

from __future__ import unicode_literals, absolute_import

import csv

import logging

import sqlalchemy as sa

from sqlalchemy import orm

from rattail import enum

from rattail.db import model, api

from rattail.gpc import GPC

            row.product = api.get_product_by_upc(session, row.upc)

        if not row.product:

            row.status_code = row.STATUS_PRODUCT_NOT_FOUND

            return

        product = row.product

        row.description = product.description

        row.size = product.size

        department = product.department

        row.department_number = department.number if department else None

        row.department_name = department.name if department else None

        category = product.category

################################################################################

#

#  Rattail -- Retail Software Framework

#  Copyright © 2010-2017 Lance Edgar

#

#  This file is part of Rattail.

"""

Handler for pricing batches

"""

from __future__ import unicode_literals, absolute_import

from sqlalchemy import orm

from rattail.db import model

from rattail.batch import BatchHandler

        Inspect a row from the source data and populate additional attributes

        for it, according to what we find in the database.

        """

        product = row.product

        assert product

        row.description = product.description

        row.size = product.size

        department = product.department

        row.department_number = department.number if department else None

        row.department_name = department.name if department else None

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

"""

Handler for purchase order batches

"""

from __future__ import unicode_literals, absolute_import

from sqlalchemy import orm

from rattail.db import model, api

from rattail.batch import BatchHandler

from rattail.time import make_utc

                return

            row.product = product

        cost = row.product.cost_for_vendor(batch.vendor) or row.product.cost

        row.upc = product.upc

        row.item_id = product.item_id

        row.description = product.description

        row.size = product.size

        if product.department:

            row.department_number = product.department.number

            row.department_name = product.department.name

        else:

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

#

################################################################################

"""

Email Bounce Handlers

"""

import os

import datetime

from email.utils import parsedate_tz, mktime_tz

from sqlalchemy import orm

from flufl.bounce import all_failures

from rattail.db import model

from rattail.core import Object

from rattail.util import load_object

    def make_links(self, session, recipient):

        url = self.config.require('tailbone', 'url.customer')

        emails = session.query(model.CustomerEmailAddress).filter_by(address=recipient)

        for email in emails:

            yield self.make_link(type="Rattail Customer",

                                 url=url.format(uuid=email.parent_uuid))

        url = self.config.require('tailbone', 'url.person')

        emails = session.query(model.PersonEmailAddress).filter_by(address=recipient)

        for email in emails:

            yield self.make_link(type="Rattail Person",

                                 url=url.format(uuid=email.parent_uuid))

"""

Views for ${model_title} batches

"""

from __future__ import unicode_literals, absolute_import

from rattail.db import model

from tailbone.views.batch import FileBatchMasterView

from .handler import ${model_name}Handler

    model_title = "${model_title} Batch"

    model_title_plural = "${model_title} Batches"

    route_prefix = 'batch.${model_name.lower()}s'

    url_prefix = '/batch/${table_name.replace('_', '-')}'

    def get_instance_title(self, batch):

    def configure_grid(self, g):

        g.joiners['vendor'] = lambda q: q.join(model.Vendor)

        g.filters['vendor'] = g.make_filter('vendor', model.Vendor.name,

                                            default_active=True, default_verb='contains')

        g.sorters['vendor'] = g.make_sorter(model.Vendor.name)

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

from __future__ import unicode_literals, absolute_import

import sys

import logging

import humanize

from rattail.time import make_utc

from rattail.util import OrderedDict

from rattail.mail import send_email

    def get_label(self, record):

        key = record.__class__.__name__.lower()

        label = getattr(self, 'label_{}'.format(key), self.label)

        return label(record)

    def label(self, record):

    def get_url(self, record):

        """

        Fetch / generate a URL for the given data record.  You should *not*

        override this method, but do :meth:`url()` instead.

        """

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

from __future__ import unicode_literals, absolute_import

import os

import datetime

import logging

from rattail.importing import BulkImporter, ToSQLAlchemy

from rattail.time import make_utc

log = logging.getLogger(__name__)

        elif isinstance(value, basestring):

            value = value.replace('\\', '\\\\')

            value = value.replace('\r', '\\r')

            value = value.replace('\n', '\\n')

            value = value.replace('\t', '\\t') # TODO: add test for this

    def flush_create_update(self):

        pass

    def flush_create_update_final(self):

        log.info("copying {} data from buffer to PostgreSQL".format(self.model_name))

        return self.config.getbool('rattail.mail', 'send_emails', default=True)

    def get_sender(self):

        """

        Returns the value for the message's ``From:`` header.

        """

        sender = self.config.get('rattail.mail', '{0}.from'.format(self.key))

        if not sender:

            sender = self.config.get('rattail.mail', 'default.from')

            if not sender:

                raise exceptions.SenderNotFound(self.key)

        return recips

    def get_prefix(self, data={}, magic=True):

        """

        Returns a string to be used as the subject prefix for the message.

        """

        prefix = self.config.get('rattail.mail', '{0}.prefix'.format(self.key))

        if not prefix:

            prefix = self.config.get('rattail.mail', 'default.prefix')

        prefix = prefix or self.default_prefix

        if magic and not self.config.production():

    def get_subject(self, data={}, render=True):

        """

        Returns the base value for the message's subject header, i.e. minus

        prefix.

        """

        subject = self.config.get('rattail.mail', '{0}.subject'.format(self.key),

                                  default=self.default_subject)

        if not subject:

            subject = self.config.get('rattail.mail', 'default.subject')

        if subject and render:

        """

        Returns the value for the message's ``Subject:`` header, i.e. the base

        subject with the prefix applied.  Note that config may provide the

        complete subject also, in which case the prefix and base subject are

        not considered.

        """

        return "{} {}".format(self.get_prefix(data), self.get_subject(data, render=render))

    def get_template(self, type_):

        """

        Locate and return the Mako email template of the given type

from __future__ import unicode_literals, absolute_import

from unittest import TestCase

import sqlalchemy as sa

from mock import patch

from fixture import TempIO

from rattail.db import model, Session, SessionBase, auth

from rattail.importing import rattail as rattail_importing

        if 'host' not in self.config.rattail_engines:

            self.config.rattail_engines['host'] = sa.create_engine('sqlite://')

        self.host_engine = self.config.rattail_engines['host']

        self.config.setdefault('rattail.db', 'keys', 'default, host')

        model = self.get_rattail_model()

        model.Base.metadata.create_all(bind=self.host_engine)

        self.host_session = Session(bind=self.host_engine)

    def teardown_rattail(self):

        super(DualRattailMixin, self).teardown_rattail()

from __future__ import unicode_literals, absolute_import

from unittest import TestCase

from mock import patch

import sqlalchemy as sa

from sqlalchemy import orm

from sqlalchemy.orm.exc import MultipleResultsFound

    def test_query(self):

        Session = orm.sessionmaker(bind=sa.create_engine('sqlite://'))

        session = Session()

        importer = saimport.FromSQLAlchemy(host_session=session,

                                           host_model_class=Widget)

                         "SELECT widgets.id AS widgets_id, widgets.description AS widgets_description \n"

                         "FROM widgets")

class TestToSQLAlchemy(TestCase):

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

#

################################################################################

"""

Vendor Catalogs

"""

from decimal import Decimal

from rattail.exceptions import RattailError

from rattail.util import load_entry_points

class CatalogParser(object):

    """

        Convert a value to an integer.

        """

        value = value.strip() or 0

        return int(value)

class CatalogParserNotFound(RattailError):

    """

    Exception raised when a vendor catalog parser is required, but cannot be

    located.

    """

    def __init__(self, key):

        self.key = key

def get_catalog_parsers():

    """

    Returns a dictionary of installed vendor catalog parser classes.

    """

################################################################################

#

#  Rattail -- Retail Software Framework

#

#  This file is part of Rattail.

#

#  Rattail is free software: you can redistribute it and/or modify it under the

#  terms of the GNU Affero General Public License as published by the Free

#  Software Foundation, either version 3 of the License, or (at your option)

#

################################################################################

"""

Vendor Invoices

"""

from decimal import Decimal

from rattail.exceptions import RattailError

from rattail.util import load_entry_points

class InvoiceParser(object):

    """

        Convert a value to an integer.

        """

        value = value.strip() or 0

        return int(value)

class InvoiceParserNotFound(RattailError):

    """

    Exception raised when a vendor invoice parser is required, but cannot be

    located.

    """

    def __init__(self, key):

        self.key = key

def get_invoice_parsers():

    """

    Returns a dictionary of installed vendor invoice parser classes.

    """