polymath-ventures-engineering-challenge/api.py at master · cyaiox/polymath-ventures-engineering-challenge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import requests
from xml.etree import ElementTree
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker


def getCategories(Base, Category, database_file):
    url = 'https://api.sandbox.ebay.com/ws/api.dll'

    headers = {
        'X-EBAY-API-CALL-NAME': 'GetCategories',
        'X-EBAY-API-APP-NAME': 'EchoBay62-5538-466c-b43b-662768d6841',
        'X-EBAY-API-CERT-NAME': '00dd08ab-2082-4e3c-9518-5f4298f296db',
        'X-EBAY-API-DEV-NAME': '16a26b1b-26cf-442d-906d-597b60c41c19',
        'X-EBAY-API-SITEID': '0',
        'X-EBAY-API-COMPATIBILITY-LEVEL': '861'
    }

    data = '<?xml version="1.0" encoding="utf-8"?> \
    <GetCategoriesRequest xmlns="urn:ebay:apis:eBLBaseComponents"> \
      <RequesterCredentials> \
        <eBayAuthToken>AgAAAA**AQAAAA**aAAAAA**PMIhVg**nY+sHZ2PrBmdj6wVnY+sEZ2PrA2dj6wFk4GhCpaCpQWdj6x9nY+seQ**L0MCAA**AAMAAA**IahulXaONmBwi/Pzhx0hMqjHhVAz9/qrFLIkfGH5wFH8Fjwj8+H5FN4NvzHaDPFf0qQtPMFUaOXHpJ8M7c2OFDJ7LBK2+JVlTi5gh0r+g4I0wpNYLtXnq0zgeS8N6KPl8SQiGLr05e9TgLRdxpxkFVS/VTVxejPkXVMs/LCN/Jr1BXrOUmVkT/4Euuo6slGyjaUtoqYMQnmBcRsK4xLiBBDtiow6YHReCJ0u8oxBeVZo3S2jABoDDO9DHLt7cS73vPQyIbdm2nP4w4BvtFsFVuaq6uMJAbFBP4F/v/U5JBZUPMElLrkXLMlkQFAB3aPvqZvpGw7S8SgL7d2s0GxnhVSbh4QAqQrQA0guK7OSqNoV+vl+N0mO24Aw8whOFxQXapTSRcy8wI8IZJynn6vaMpBl5cOuwPgdLMnnE+JvmFtQFrxa+k/9PRoVFm+13iGoue4bMY67Zcbcx65PXDXktoM3V+sSzSGhg5M+R6MXhxlN3xYfwq8vhBQfRlbIq+SU2FhicEmTRHrpaMCk4Gtn8CKNGpEr1GiNlVtbfjQn0LXPp7aYGgh0A/b8ayE1LUMKne02JBQgancNgMGjByCIemi8Dd1oU1NkgICFDbHapDhATTzgKpulY02BToW7kkrt3y6BoESruIGxTjzSVnSAbGk1vfYsQRwjtF6BNbr5Goi52M510DizujC+s+lSpK4P0+RF9AwtrUpVVu2PP8taB6FEpe39h8RWTM+aRDnDny/v7wA/GkkvfGhiioCN0z48</eBayAuthToken> \
      </RequesterCredentials> \
    <CategorySiteID>0</CategorySiteID>\
      <DetailLevel>ReturnAll</DetailLevel>\
    </GetCategoriesRequest>'

    response = requests.post(url, headers=headers, data=data)

    saveCategories(response.text, Base, Category, database_file)


def saveCategories(response, Base, Category, database_file):
    tree = ElementTree.fromstring(response)
    engine = create_engine('sqlite:///%s' % database_file)
    parents = {}
    Base.metadata.bind = engine

    DBSession = sessionmaker(bind=engine)

    session = DBSession()

    for child in tree:
        if child.tag == '{urn:ebay:apis:eBLBaseComponents}CategoryArray':
            for category in ElementTree.ElementTree(element=child).iterfind('{urn:ebay:apis:eBLBaseComponents}Category'):
                new_category = Category()
                for attribute in ElementTree.ElementTree(element=category).iter():
                    if attribute.tag == '{urn:ebay:apis:eBLBaseComponents}CategoryID':
                        new_category.id = int(attribute.text)

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}CategoryName':
                        new_category.name = attribute.text

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}CategoryLevel':
                        new_category.level = int(attribute.text)

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}CategoryParentID':
                        new_category.parent_id = int(attribute.text)

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}BestOfferEnabled':
                        new_category.is_best_offer_enabled = True

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}AutoPayEnabled':
                        new_category.is_auto_pay_enabled = True

                    elif attribute.tag == '{urn:ebay:apis:eBLBaseComponents}LeafCategory':
                        new_category.is_leaf_category = True

                if new_category.level == 1 and not new_category.is_leaf_category:
                    parents[new_category.level] = new_category
                    session.add(parents[new_category.level])

                elif new_category.level == 1:
                    session.add(new_category)

                else:
                    if not new_category.is_leaf_category:
                        parents[new_category.level] = new_category

                    new_category.parent = parents[(new_category.level - 1)]
                    session.add(new_category)

                    parents[(new_category.level - 1)].children.append(new_category)
                    session.add(parents[(new_category.level - 1)])

                session.commit()