Store DART entity IDs

suminb · May 30, 2017 · bc843f3 · bc843f3
1 parent 5ef5e78
commit bc843f3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/finance/models.py b/finance/models.py
@@ -470,6 +470,7 @@ class DartReport(CRUDMixin, db.Model):
 
     registered_at = db.Column(db.DateTime(timezone=False))
     title = db.Column(db.String)
+    entity_id = db.Column(db.Integer)
     entity = db.Column(db.String)
     reporter = db.Column(db.String)
     content = db.Column(db.Text)
diff --git a/finance/providers/dart.py b/finance/providers/dart.py
@@ -107,7 +107,7 @@ class Report(object):
     id = Integer()
     registered_at = DateTime(date_format='%Y.%m.%d')
     title = String()
-    entity_crp = Integer()  # 전자공시 CRP code
+    entity_id = Integer()  # 전자공시 CRP code
     entity = String()
     reporter = String()
     content = String()
@@ -116,7 +116,7 @@ def __init__(self, **kwargs):
         self.id = kwargs['rcp_no']
         self.registered_at = kwargs['rcp_dm']
         self.title = kwargs['rptNm']
-        self.entity_crp = kwargs['dsm_crp_cik']
+        self.entity_id = kwargs['dsm_crp_cik']
         self.entity = kwargs['ifm_nm']
         self.reporter = kwargs['ifm_nm2']
         self.content = kwargs['reportBody']
@@ -127,7 +127,7 @@ def __repr__(self):
             self.registered_at.strftime('%Y-%m-%d'), self.entity)
 
     def __iter__(self):
-        attrs = ['id', 'registered_at', 'title', 'entity_crp', 'entity',
+        attrs = ['id', 'registered_at', 'title', 'entity_id', 'entity',
                 'reporter', 'content']
         for attr in attrs:
             yield attr, getattr(self, attr)