diff options
Diffstat (limited to 'bin/common/srtool_common.py')
-rwxr-xr-x | bin/common/srtool_common.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/bin/common/srtool_common.py b/bin/common/srtool_common.py index 46996aec..4cba2dc2 100755 --- a/bin/common/srtool_common.py +++ b/bin/common/srtool_common.py @@ -360,7 +360,7 @@ def score_new_cves(cve_filter): packages = ?, score_date = ? WHERE id = ?''' - cur_write.execute(sql, (recommend, recommend_list, cve_packages, time_now.strftime('%Y-%m-%d %H:%M:%S'), cve[ORM.CVE_ID])) + cur_write.execute(sql, (recommend, recommend_list, cve_packages, time_now.strftime(ORM.DATASOURCE_DATETIME_FORMAT), cve[ORM.CVE_ID])) write_count += 1 if verbose: print(" %d:%s:%s" % (recommend,recommend_list,cve_packages)) @@ -499,9 +499,12 @@ def gen_schema_header(): fd.write(" %s_%s = '%s'\n" % ('DATASOURCE','FREQUENCY_STR', \ 'Minute,Hourly,Daily,Weekly,Monthly,OnDemand,OnStartup' \ )) + fd.write(" %s_%s = '%s'\n" % ('DATASOURCE','DATE_FORMAT','%Y-%m-%d')) + fd.write(" %s_%s = '%s'\n" % ('DATASOURCE','DATETIME_FORMAT','%Y-%m-%d %H:%M:%S')) fd.write("\n\n") fd.write(" # General routine to return string name of a constant (e.g. 'DATASOURCE_FREQUENCY_STR')\n") + fd.write(" @staticmethod\n") fd.write(" def get_orm_string(value,string_set):\n") fd.write(" string_list = string_set.split(',')\n") fd.write(" string_count = len(string_list)\n") |