Merge pull request #77 from Bidaya0/scamaven

sca_maven
HXSecurity · Jan 12, 2022 · 3b08080 · 3b08080
2 parents 3fa5924 + d82f232
commit 3b08080
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/dongtai/models/sca_maven_db.py b/dongtai/models/sca_maven_db.py
@@ -8,14 +8,25 @@
 from dongtai.utils.settings import get_managed
 
 
+class ImportFrom(models.IntegerChoices):
+    SYSTEM = 1
+    USER = 2
+    __empty__ = 2
+
+
 class ScaMavenDb(models.Model):
     group_id = models.CharField(max_length=255, blank=True, null=True)
     atrifact_id = models.CharField(max_length=255, blank=True, null=True)
     version = models.CharField(max_length=255, blank=True, null=True)
-    sha_1 = models.CharField(unique=True, max_length=255, blank=True, null=True)
+    sha_1 = models.CharField(unique=True,
+                             max_length=255,
+                             blank=True,
+                             null=True)
     package_name = models.CharField(max_length=255, blank=True, null=True)
     aql = models.CharField(max_length=255, blank=True, null=True)
     license = models.CharField(max_length=255, blank=True, null=True)
+    import_from = models.IntegerField(choices=ImportFrom.choices,
+                                      default=ImportFrom.USER)
 
     class Meta:
         managed = get_managed()