Adding Support for Custom Segments in NetSuite (#74)

netsuitesdk/api/custom_segments.py

@@ -0,0 +1,18 @@
+from .base import ApiBase
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class CustomSegments(ApiBase):
+    def __init__(self, ns_client):
+        ApiBase.__init__(self, ns_client=ns_client, type_name='CustomSegment')
+
+    def get_all(self):
+        return self._get_all()
+
+    def get_all_generator(self, page_size=20):
+        """
+        Returns a generator which is more efficient memory-wise
+        """
+        return self._get_all_generator()

netsuitesdk/connection.py

@@ -17,6 +17,7 @@
 from .api.projects import Projects
 from .api.expense_categories import ExpenseCategory
 from .api.custom_lists import CustomLists
+from .api.custom_segments import CustomSegments
 from .api.custom_record_types import CustomRecordTypes
 from .api.custom_records import CustomRecords
 from .api.vendor_payments import VendorPayments
@@ -57,6 +58,7 @@ def __init__(self, account, consumer_key, consumer_secret, token_key, token_secr
         self.files = Files(ns_client)
         self.expense_categories = ExpenseCategory(ns_client)
         self.custom_lists = CustomLists(ns_client)
+        self.custom_segments = CustomSegments(ns_client)
         self.custom_records = CustomRecords(ns_client)
         self.custom_record_types = CustomRecordTypes(ns_client)
         self.customers = Customers(ns_client)

netsuitesdk/internal/constants.py

@@ -23,6 +23,7 @@
     'customerMessage',
     'customerStatus',
     'customList',
+    'customSegment',
     'customRecord',
     'department',
     'employee',

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='netsuitesdk',
-    version='2.8.0',
+    version='2.9.0',
     author='Siva Narayanan',
     author_email='[email protected]',
     description='Python SDK for accessing the NetSuite SOAP webservice',