Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
beat.editor
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Model registry
Operate
Environments
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
beat
beat.editor
Commits
2bfbacdf
Commit
2bfbacdf
authored
6 years ago
by
Samuel GAIST
Browse files
Options
Downloads
Patches
Plain Diff
[utils] Code cleanup based on black and flake8
parent
a38c588a
No related branches found
No related tags found
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
beat/editor/utils.py
+25
-17
25 additions, 17 deletions
beat/editor/utils.py
with
25 additions
and
17 deletions
beat/editor/utils.py
+
25
−
17
View file @
2bfbacdf
...
@@ -32,10 +32,11 @@ import sys
...
@@ -32,10 +32,11 @@ import sys
import
jinja2
import
jinja2
import
logging
import
logging
logger
=
logging
.
getLogger
(
__name__
)
logger
=
logging
.
getLogger
(
__name__
)
ENV
=
jinja2
.
Environment
(
loader
=
jinja2
.
PackageLoader
(
__name__
,
'
templates
'
))
ENV
=
jinja2
.
Environment
(
loader
=
jinja2
.
PackageLoader
(
__name__
,
"
templates
"
))
"""
Jinja2 environment for loading our templates
"""
"""
Jinja2 environment for loading our templates
"""
...
@@ -55,8 +56,8 @@ def generate_database(views=None):
...
@@ -55,8 +56,8 @@ def generate_database(views=None):
"""
"""
views
=
views
or
[
'
View
'
]
views
=
views
or
[
"
View
"
]
template
=
ENV
.
get_template
(
'
database.jinja2
'
)
template
=
ENV
.
get_template
(
"
database.jinja2
"
)
return
template
.
render
(
views
=
views
)
return
template
.
render
(
views
=
views
)
...
@@ -78,7 +79,7 @@ def generate_library(uses=None):
...
@@ -78,7 +79,7 @@ def generate_library(uses=None):
"""
"""
uses
=
uses
or
{}
uses
=
uses
or
{}
template
=
ENV
.
get_template
(
'
library.jinja2
'
)
template
=
ENV
.
get_template
(
"
library.jinja2
"
)
return
template
.
render
(
uses
=
uses
)
return
template
.
render
(
uses
=
uses
)
...
@@ -97,7 +98,7 @@ def generate_algorithm(contents):
...
@@ -97,7 +98,7 @@ def generate_algorithm(contents):
"""
"""
template
=
ENV
.
get_template
(
'
algorithm.jinja2
'
)
template
=
ENV
.
get_template
(
"
algorithm.jinja2
"
)
return
template
.
render
(
contents
=
contents
)
return
template
.
render
(
contents
=
contents
)
...
@@ -117,7 +118,7 @@ def generate_plotter(uses):
...
@@ -117,7 +118,7 @@ def generate_plotter(uses):
"""
"""
uses
=
uses
or
{}
uses
=
uses
or
{}
template
=
ENV
.
get_template
(
'
plotter.jinja2
'
)
template
=
ENV
.
get_template
(
"
plotter.jinja2
"
)
return
template
.
render
(
uses
=
uses
)
return
template
.
render
(
uses
=
uses
)
...
@@ -130,9 +131,11 @@ TEMPLATE_FUNCTION = dict(
...
@@ -130,9 +131,11 @@ TEMPLATE_FUNCTION = dict(
"""
Functions for template instantiation within beat.editor
"""
"""
Functions for template instantiation within beat.editor
"""
class
PythonFileAlreadyExistsError
(
Exception
):
class
PythonFileAlreadyExistsError
(
Exception
):
pass
pass
def
generate_python_template
(
entity
,
name
,
confirm
,
config
,
**
kwargs
):
def
generate_python_template
(
entity
,
name
,
confirm
,
config
,
**
kwargs
):
"""
Generates a template for a BEAT entity with the given named arguments
"""
Generates a template for a BEAT entity with the given named arguments
...
@@ -148,20 +151,21 @@ def generate_python_template(entity, name, confirm, config, **kwargs):
...
@@ -148,20 +151,21 @@ def generate_python_template(entity, name, confirm, config, **kwargs):
"""
"""
resource_path
=
os
.
path
.
join
(
config
.
path
,
entity
)
resource_path
=
os
.
path
.
join
(
config
.
path
,
entity
)
file_path
=
os
.
path
.
join
(
resource_path
,
name
)
+
'
.py
'
file_path
=
os
.
path
.
join
(
resource_path
,
name
)
+
"
.py
"
if
not
confirm
and
os
.
path
.
isfile
(
file_path
):
if
not
confirm
and
os
.
path
.
isfile
(
file_path
):
# python file already exists
# python file already exists
raise
PythonFileAlreadyExistsError
raise
PythonFileAlreadyExistsError
s
=
TEMPLATE_FUNCTION
[
entity
](
**
kwargs
)
s
=
TEMPLATE_FUNCTION
[
entity
](
**
kwargs
)
with
open
(
file_path
,
'
w
'
)
as
f
:
f
.
write
(
s
)
with
open
(
file_path
,
"
w
"
)
as
f
:
f
.
write
(
s
)
return
s
return
s
def
setup_logger
(
name
,
verbosity
):
def
setup_logger
(
name
,
verbosity
):
'''
Sets up the logging of a script
"""
Sets up the logging of a script
Parameters:
Parameters:
...
@@ -173,18 +177,20 @@ def setup_logger(name, verbosity):
...
@@ -173,18 +177,20 @@ def setup_logger(name, verbosity):
warnings and informational messages and, finally, ``3``, all types of
warnings and informational messages and, finally, ``3``, all types of
messages including debugging ones.
messages including debugging ones.
'''
"""
logger
=
logging
.
getLogger
(
name
)
logger
=
logging
.
getLogger
(
name
)
formatter
=
logging
.
Formatter
(
"
%(name)s@%(asctime)s -- %(levelname)s:
"
\
formatter
=
logging
.
Formatter
(
"
%(message)s
"
)
"
%(name)s@%(asctime)s -- %(levelname)s:
"
"
%(message)s
"
)
_warn_err
=
logging
.
StreamHandler
(
sys
.
stderr
)
_warn_err
=
logging
.
StreamHandler
(
sys
.
stderr
)
_warn_err
.
setFormatter
(
formatter
)
_warn_err
.
setFormatter
(
formatter
)
_warn_err
.
setLevel
(
logging
.
WARNING
)
_warn_err
.
setLevel
(
logging
.
WARNING
)
class
_InfoFilter
:
class
_InfoFilter
:
def
filter
(
self
,
record
):
return
record
.
levelno
<=
logging
.
INFO
def
filter
(
self
,
record
):
return
record
.
levelno
<=
logging
.
INFO
_debug_info
=
logging
.
StreamHandler
(
sys
.
stdout
)
_debug_info
=
logging
.
StreamHandler
(
sys
.
stdout
)
_debug_info
.
setFormatter
(
formatter
)
_debug_info
.
setFormatter
(
formatter
)
...
@@ -194,10 +200,12 @@ def setup_logger(name, verbosity):
...
@@ -194,10 +200,12 @@ def setup_logger(name, verbosity):
logger
.
addHandler
(
_debug_info
)
logger
.
addHandler
(
_debug_info
)
logger
.
addHandler
(
_warn_err
)
logger
.
addHandler
(
_warn_err
)
logger
.
setLevel
(
logging
.
ERROR
)
logger
.
setLevel
(
logging
.
ERROR
)
if
verbosity
==
1
:
logger
.
setLevel
(
logging
.
WARNING
)
if
verbosity
==
1
:
elif
verbosity
==
2
:
logger
.
setLevel
(
logging
.
INFO
)
logger
.
setLevel
(
logging
.
WARNING
)
elif
verbosity
>=
3
:
logger
.
setLevel
(
logging
.
DEBUG
)
elif
verbosity
==
2
:
logger
.
setLevel
(
logging
.
INFO
)
elif
verbosity
>=
3
:
logger
.
setLevel
(
logging
.
DEBUG
)
return
logger
return
logger
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment