format

Author: kzrnm

onlinejudge_verify/languages/csharp.py

@@ -1,15 +1,14 @@
 # Python Version: 3.x
+import distutils.version
 import functools
 import json
 import os
 import pathlib
+import shutil
 import subprocess
+import xml.etree.ElementTree as ET
 from logging import getLogger
-from turtle import st
 from typing import *
-import distutils.version
-import shutil
-import xml.etree.ElementTree as ET
 
 import onlinejudge_verify.languages.special_comments as special_comments
 from onlinejudge_verify.languages.models import Language, LanguageEnvironment
@@ -34,15 +33,13 @@ def _check_dotnet_version() -> None:
 @functools.lru_cache(maxsize=1)
 def _check_expander_console() -> None:
     if not shutil.which('dotnet-source-expand'):
-        raise RuntimeError(
-            '`dotnet-source-expand` not in $PATH. Run `dotnet tool install -g SourceExpander.Console`')
+        raise RuntimeError('`dotnet-source-expand` not in $PATH. Run `dotnet tool install -g SourceExpander.Console`')
     command = ['dotnet-source-expand', 'version']
     logger.info('$ %s', ' '.join(command))
     res = subprocess.check_output(command).decode().strip()
     logger.info('dotnet-source-expand version: %s', res)
     if distutils.version.LooseVersion(res) < distutils.version.LooseVersion("5"):
-        raise RuntimeError(
-            '`dotnet-source-expand` version must be 5.0.0 or newer. Update SourceExpander.Console. `dotnet tool update -g SourceExpander.Console`')
+        raise RuntimeError('`dotnet-source-expand` version must be 5.0.0 or newer. Update SourceExpander.Console. `dotnet tool update -g SourceExpander.Console`')
 
 
 class EmbeddedLibrary:
@@ -58,7 +55,7 @@ def __repr__(self) -> str:
 def _list_embedded(csproj_path: pathlib.Path) -> List[EmbeddedLibrary]:
     _check_expander_console()
     if csproj_path is None or csproj_path.suffix != ".csproj":
-        raise None
+        raise RuntimeError('csproj_path must be .csproj')
     command = ['dotnet-source-expand', 'library-list', str(csproj_path)]
     logger.info('$ %s', ' '.join(command))
 
@@ -68,10 +65,7 @@ def enumerate_library(lines: List[str]):
             if len(sp) >= 2:
                 yield EmbeddedLibrary(sp[0], sp[1])
 
-    res = list(enumerate_library(
-        subprocess.check_output(
-            command, encoding='utf-8'
-        ).strip().splitlines()))
+    res = list(enumerate_library(subprocess.check_output(command, encoding='utf-8').strip().splitlines()))
     logger.debug('libraries: %s', res)
     return res
 
@@ -83,8 +77,7 @@ def _check_embedded_existing(csproj_path: pathlib.Path) -> None:
     subprocess.check_output(command)
     l = _list_embedded(csproj_path)
     if len(l) == 0:
-        raise RuntimeError(
-            'Library needs SourceExpander.Embedder')
+        raise RuntimeError('Library needs SourceExpander.Embedder')
 
 
 def _check_env(path: pathlib.Path):
@@ -97,8 +90,7 @@ def _check_env(path: pathlib.Path):
 def _check_no_embedder(csproj_path: pathlib.Path) -> None:
     root = ET.parse(csproj_path).getroot()
     if root.find('.//PackageReference[@Include="SourceExpander.Embedder"]'):
-        logger.error(
-            " Test project(%s) has `SourceExpander.Embedder` reference. Libraries and tests should not be in same project.", str(csproj_path))
+        logger.error(" Test project(%s) has `SourceExpander.Embedder` reference. Libraries and tests should not be in same project.", str(csproj_path))
 
 
 @functools.lru_cache(maxsize=None)
@@ -122,8 +114,7 @@ def _expand_code_dict(csproj_path: pathlib.Path) -> Dict[pathlib.Path, str]:
     command = ['dotnet-source-expand', 'expand-all', str(csproj_path)]
     logger.info('$ %s', ' '.join(command))
     json_res = subprocess.check_output(command)
-    return {pathlib.Path(t['FilePath']): t['ExpandedCode']
-            for t in json.loads(json_res)}
+    return {pathlib.Path(t['FilePath']): t['ExpandedCode'] for t in json.loads(json_res)}
 
 
 @functools.lru_cache(maxsize=None)
@@ -150,14 +141,12 @@ def _dependency_info_list(csproj_path: pathlib.Path) -> List[DependencyInfo]:
     _check_expander_console()
     _check_embedded_existing(csproj_path)
     if csproj_path is None or csproj_path.suffix != ".csproj":
-        raise None
+        raise RuntimeError('csproj_path must be .csproj')
 
     command = ['dotnet-source-expand', 'dependency', '-p', str(csproj_path)]
     logger.info('$ %s', ' '.join(command))
     res = subprocess.check_output(command)
-    return json.loads(
-        res,
-        object_hook=lambda d: DependencyInfo(d['FileName'], d['Dependencies'], set(d['TypeNames'])))
+    return json.loads(res, object_hook=lambda d: DependencyInfo(d['FileName'], d['Dependencies'], set(d['TypeNames'])))
 
 
 @functools.lru_cache(maxsize=None)
@@ -174,62 +163,57 @@ def _dependency_info_dict(csproj_path: pathlib.Path) -> Dict[pathlib.Path, Depen
 def _list_dependencies(path: pathlib.Path) -> List[pathlib.Path]:
     path = path.resolve()
     depinfo = _dependency_info_dict(_resolve_csproj(path))
-    return [p
-            for p in (
-                pathlib.Path(dep)
-                for dep in depinfo[path].dependencies
-            ) if p.exists()]
+    return [p for p in (pathlib.Path(dep) for dep in depinfo[path].dependencies) if p.exists()]
 
 
 @functools.lru_cache(maxsize=None)
-def _get_target_framework(csproj_path: pathlib.Path) -> bytes:
+def _get_target_framework(csproj_path: pathlib.Path) -> str:
     root = ET.parse(csproj_path).getroot()
     target = root.findtext('.//TargetFramework')
+    if target is None:
+        raise RuntimeError('<TargetFramework> is not found')
     return target
 
 
 class CSharpLanguageEnvironment(LanguageEnvironment):
     @staticmethod
     def _create_runner_project(code: bytes, target_framework: str, output_dir):
         os.makedirs(str(output_dir), exist_ok=True)
-        with open(output_dir/'runner.csproj', 'w') as f:
+        with open(output_dir / 'runner.csproj', 'w') as f:
             f.write('''<Project Sdk="Microsoft.NET.Sdk">
   <PropertyGroup>
     <OutputType>Exe</OutputType>
     <TargetFramework>{}</TargetFramework>
   </PropertyGroup>
 </Project>'''.format(target_framework))
 
-        with open(output_dir/'main.cs', 'wb') as f:
+        with open(output_dir / 'main.cs', 'wb') as f:
             f.write(code)
 
     def compile(self, path: pathlib.Path, *, basedir: pathlib.Path, tempdir: pathlib.Path) -> None:
         path = path.resolve()
-        output_dir = tempdir/'dotnet'
+        output_dir = tempdir / 'dotnet'
         _check_env(path)
         target_framework = _get_target_framework(_resolve_csproj(path))
         logger.info('build: TargetFramework = %s', target_framework)
-        self._create_runner_project(
-            _expand_code(path), target_framework, output_dir)
+        self._create_runner_project(_expand_code(path), target_framework, output_dir)
 
-        command = ['dotnet', 'build', str(output_dir/'runner.csproj'),
-                   '-c', 'Release',
-                   '-o', str(output_dir/'bin')]
+        command = ['dotnet', 'build', str(output_dir / 'runner.csproj'), '-c', 'Release', '-o', str(output_dir / 'bin')]
         logger.info('$ %s', ' '.join(command))
         subprocess.check_output(command)
 
     def get_execute_command(self, path: pathlib.Path, *, basedir: pathlib.Path, tempdir: pathlib.Path) -> List[str]:
         path = path.resolve()
-        output_dir = tempdir/'dotnet'
+        output_dir = tempdir / 'dotnet'
         path = path.resolve()
         _check_env(path)
-        return [str(output_dir/'bin'/'runner')]
+        return [str(output_dir / 'bin' / 'runner')]
 
 
 class CSharpLanguage(Language):
     def list_attributes(self, path: pathlib.Path, *, basedir: pathlib.Path) -> Dict[str, Any]:
         path = path.resolve()
-        attributes = special_comments.list_special_comments(path)
+        attributes: Dict[str, Any] = special_comments.list_special_comments(path)
         attributes.setdefault('links', [])
         attributes['links'].extend(special_comments.list_embedded_urls(path))
         return attributes