Skip to content

GitLab

"lib/llm/src/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "44bde250656f68864dcd4942d8648454afc928fa"
Commit a44bbebb authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Add argument check to CAMEO script

parent b18026fd
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
scripts/download_cameo.py
View file @ a44bbebb
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
import argparse import argparse
import json import json
import os import os
import re
import requests import requests
from openfold.data import mmcif_parsing from openfold.data import mmcif_parsing
...@@ -73,6 +74,7 @@ def main(args): ...@@ -73,6 +74,7 @@ def main(args):
with open(os.path.join(data_dir_path, cif_filename), "w") as fp: with open(os.path.join(data_dir_path, cif_filename), "w") as fp:
fp.write(pdb_file) fp.write(pdb_file)
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
...@@ -95,4 +97,8 @@ if __name__ == '__main__': ...@@ -95,4 +97,8 @@ if __name__ == '__main__':
if(args.period not in VALID_PERIODS): if(args.period not in VALID_PERIODS):
raise ValueError(f"Invalid period. Choose from {VALID_PERIODS}") raise ValueError(f"Invalid period. Choose from {VALID_PERIODS}")
date_regex = re.compile("^[0-9]{4}-[0-9]{2}-[0-9]{2}$")
if(not date_regex.match(args.end_date)):
raise ValueError(f"Invalid end_date: {args.end_date}. Use YYYY-MM-DD format")
main(args) main(args)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment