Skip to content

Commit

Permalink
Working with packages
Browse files Browse the repository at this point in the history
  • Loading branch information
William Strecker-Kellogg committed Apr 14, 2021
1 parent 09566fa commit f009aee
Show file tree
Hide file tree
Showing 10 changed files with 71 additions and 64 deletions.
1 change: 1 addition & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,4 @@ include README.md

graft conf
graft cron
graft static
2 changes: 2 additions & 0 deletions sdccjupyter/forms/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from .icform import icform
from .otherforms import *
64 changes: 5 additions & 59 deletions sdccjupyter/forms.py → sdccjupyter/forms/icform.py
Original file line number Diff line number Diff line change
@@ -1,64 +1,10 @@
from jinja2 import Template
from tornado.log import app_log

from .formspawners import ParamForm
from ..formspawners import ParamForm

import sqlite3

__all__ = ['CFNForm', 'NSLSForm', 'KNLForm', 'SDCCForm', 'ICForm']


class CFNForm(ParamForm):

source = 'static/cfn.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
intify = {'req_memory', 'req_nprocs'}
data = {k: int(v) if v in intify else v for k, v in data.items()}
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class NSLSForm(ParamForm):
source = 'static/nsls.html'

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class KNLForm(ParamForm):

source = 'static/knl.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class SDCCForm(ParamForm):

source = 'static/sdcc.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class ICForm(ParamForm):

Expand All @@ -76,7 +22,7 @@ class ICForm(ParamForm):
def slurm_time_to_min(timestr):
# Strings look like '1-12:30:00' for 1 day 12 1/2 hours
s_m_hd = timestr.split(':')
days, hours, min, sec = [0] * 4
days, hours, mins, sec = [0] * 4

if len(s_m_hd) > 2:
hd = s_m_hd.pop(0)
Expand All @@ -85,8 +31,8 @@ def slurm_time_to_min(timestr):
else:
hours = int(hd)
if len(s_m_hd) > 1:
min = int(s_m_hd.pop(0))
return days * 24 * 60 + hours * 60 + min
mins = int(s_m_hd.pop(0))
return days * 24 * 60 + hours * 60 + mins

def massage_options(self, formdata):
data = super().massage_options(formdata)
Expand Down Expand Up @@ -121,4 +67,4 @@ def generate(self):
}
db.close()

return Template(super().generate()).render(**vars)
return Template(super().generate()).render(**vars)
60 changes: 60 additions & 0 deletions sdccjupyter/forms/otherforms.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
from jinja2 import Template
from tornado.log import app_log

from .formspawners import ParamForm

import sqlite3

__all__ = ['CFNForm', 'NSLSForm', 'KNLForm', 'SDCCForm']


class CFNForm(ParamForm):

source = 'static/cfn.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
intify = {'req_memory', 'req_nprocs'}
data = {k: int(v) if v in intify else v for k, v in data.items()}
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class NSLSForm(ParamForm):
source = 'static/nsls.html'

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class KNLForm(ParamForm):

source = 'static/knl.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()


class SDCCForm(ParamForm):

source = 'static/sdcc.html'

def massage_options(self, formdata):
data = super().massage_options(formdata)
data['req_runtime'] = '%d:00' % int(data['req_runtime'])
return data

def generate(self):
app_log.info("Generating form from: %s", self)
return Template(super().generate()).render()
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
8 changes: 3 additions & 5 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,7 @@

import sys

from distutils.core import setup

from setuptools import setup, find_packages

VERSION = '0.2.0'

Expand All @@ -24,9 +23,8 @@
def main():
setup(
name = 'sdccjupyter',
packages = ['sdccjupyter'],
package_data = {'sdccjupyter': ['conf/*.cfg', 'cron/*']},
# include_package_data = True,
packages = find_packages(),
package_data = {'sdccjupyter': ['conf/*.cfg', 'cron/*', 'static/*.html']},
version = VERSION,
description = """SDCC Jupyter Spawners""",
long_description = "",
Expand Down

0 comments on commit f009aee

Please sign in to comment.