@@ -85,18 +85,6 @@ def make_object(self, data):
8585 return ContactInfo (** data )
8686
8787
88- class AccountSchema (BaseSchema ):
89- name = fields .Str ()
90- external_id = fields .Str (allow_none = True )
91- external_uid = fields .Str (allow_none = True )
92- contact_info = fields .Nested (ContactInfoSchema )
93-
94- @post_load
95- def make_object (self , data ):
96- from connect .models import Account
97- return Account (** data )
98-
99-
10088class ValueChoiceSchema (Schema ):
10189 value = fields .Str ()
10290 label = fields .Str ()
@@ -396,27 +384,27 @@ def make_object(self, data):
396384 return Template (** data )
397385
398386
399- class TierSchema (BaseSchema ):
387+ class TierAccountSchema (BaseSchema ):
400388 name = fields .Str ()
401389 contact_info = fields .Nested (ContactInfoSchema )
402390 external_id = fields .Str ()
403391 external_uid = fields .Str ()
404392
405393 @post_load
406394 def make_object (self , data ):
407- from connect .models import Tier
408- return Tier (** data )
395+ from connect .models import TierAccount
396+ return TierAccount (** data )
409397
410398
411- class TiersSchema (Schema ):
412- customer = fields .Nested (TierSchema )
413- tier1 = fields .Nested (TierSchema )
414- tier2 = fields .Nested (TierSchema )
399+ class TierAccountsSchema (Schema ):
400+ customer = fields .Nested (TierAccountSchema )
401+ tier1 = fields .Nested (TierAccountSchema )
402+ tier2 = fields .Nested (TierAccountSchema )
415403
416404 @post_load
417405 def make_object (self , data ):
418- from connect .models import Tiers
419- return Tiers (** data )
406+ from connect .models import TierAccounts
407+ return TierAccounts (** data )
420408
421409
422410class ConnectionSchema (BaseSchema ):
@@ -442,7 +430,7 @@ class AssetSchema(BaseSchema):
442430 )
443431 items = fields .Nested (ItemSchema , many = True )
444432 params = fields .Nested (ParamSchema , many = True )
445- tiers = fields .Nested (TiersSchema )
433+ tiers = fields .Nested (TierAccountsSchema )
446434
447435 @post_load
448436 def make_object (self , data ):
@@ -471,7 +459,7 @@ def make_object(self, data):
471459
472460class TierConfigSchema (BaseSchema ):
473461 name = fields .Str ()
474- account = fields .Nested (AccountSchema )
462+ account = fields .Nested (TierAccountSchema )
475463 product = fields .Nested (ProductSchema )
476464 tier_level = fields .Int ()
477465 connection = fields .Nested (ConnectionSchema )
@@ -579,3 +567,28 @@ class UsageRecordSchema(BaseSchema):
579567 def make_object (self , data ):
580568 from connect .models import UsageRecord
581569 return UsageRecord (** data )
570+
571+
572+ class ConversationMessageSchema (BaseSchema ):
573+ conversation = fields .Str ()
574+ created = fields .DateTime ()
575+ creator = fields .Nested (UserSchema )
576+ text = fields .Str ()
577+
578+ @post_load
579+ def make_object (self , data ):
580+ from connect .models import ConversationMessage
581+ return ConversationMessage (** data )
582+
583+
584+ class ConversationSchema (BaseSchema ):
585+ instance_id = fields .Str ()
586+ created = fields .DateTime ()
587+ topic = fields .Str ()
588+ messages = fields .Nested (ConversationMessageSchema , many = True )
589+ creator = fields .Nested (UserSchema )
590+
591+ @post_load
592+ def make_object (self , data ):
593+ from connect .models import Conversation
594+ return Conversation (** data )
0 commit comments