From 5c7b9c0cbec4f39262a047009a2f51004c97ff77 Mon Sep 17 00:00:00 2001
From: yodeng <yodeng@tju.edu.cn>
Date: Wed, 26 Apr 2023 14:38:05 +0800
Subject: [PATCH] update some object name

---
 runjob/__init__.py         | 4 ++--
 setup.py                   | 6 +++---
 src/__init__.py            | 4 ++--
 src/jobstat.py             | 5 ++---
 src/qsub.py                | 6 +++---
 src/{sge_run.py => run.py} | 8 ++++----
 src/utils.py               | 2 +-
 7 files changed, 17 insertions(+), 18 deletions(-)
 rename src/{sge_run.py => run.py} (99%)

diff --git a/runjob/__init__.py b/runjob/__init__.py
index 85c6b73..ca044c6 100644
--- a/runjob/__init__.py
+++ b/runjob/__init__.py
@@ -1,7 +1,7 @@
-from .utils import Mylog as log
+from .utils import getlog as log
 from .utils import JobQueue, QsubError, JobFailedError
 from .loger import Formatter
 from ._version import __version__
 from .qsub import qsub as runjob
-from .sge_run import RunSge as runsge
+from .run import RunJob as runsge
 from .config import load_config as Config
diff --git a/setup.py b/setup.py
index eba73fe..1aa8420 100644
--- a/setup.py
+++ b/setup.py
@@ -108,9 +108,9 @@ def _entrys(self):
             '%s = %s.qsub:main' % (self.name, self.name),
             '%s = %s.jobstat:main' % ("qs", self.name),
             '%s = %s.jobstat:batchStat' % ("qcs", self.name),
-            '%s = %s.sge_run:main' % ("runsge", self.name),
-            '%s = %s.sge_run:main' % ("runshell", self.name),
-            '%s = %s.sge_run:main' % ("runbatch", self.name),
+            '%s = %s.run:main' % ("runsge", self.name),
+            '%s = %s.run:main' % ("runshell", self.name),
+            '%s = %s.run:main' % ("runbatch", self.name),
         ]
         return eps
 
diff --git a/src/__init__.py b/src/__init__.py
index 85c6b73..ca044c6 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1,7 +1,7 @@
-from .utils import Mylog as log
+from .utils import getlog as log
 from .utils import JobQueue, QsubError, JobFailedError
 from .loger import Formatter
 from ._version import __version__
 from .qsub import qsub as runjob
-from .sge_run import RunSge as runsge
+from .run import RunJob as runsge
 from .config import load_config as Config
diff --git a/src/jobstat.py b/src/jobstat.py
index 1df3f6c..9553b54 100644
--- a/src/jobstat.py
+++ b/src/jobstat.py
@@ -5,7 +5,6 @@
        qcs --help
 '''
 
-from __future__ import print_function
 import os
 import re
 import sys
@@ -20,7 +19,7 @@
 from collections import defaultdict
 
 from .utils import *
-from .bc_stat import *
+from .stat_bc import *
 from .cluster import *
 from .job import Jobfile
 from ._version import __version__
@@ -267,7 +266,7 @@ def batchStat():
     if access_key_secret is None or access_key_id is None:
         sys.exit("No access to connect OSS")
     client = Client(region, access_key_id, access_key_secret)
-    logger = Mylog(name=__name__)
+    logger = getlog(name=__name__)
     user = getpass.getuser()
     if args.job:
         try:
diff --git a/src/qsub.py b/src/qsub.py
index 4f6049d..ec4d725 100644
--- a/src/qsub.py
+++ b/src/qsub.py
@@ -4,11 +4,11 @@
 from . import dag
 from .utils import *
 from .job import Jobfile
-from .sge_run import RunSge
+from .run import RunJob
 from .config import load_config
 
 
-class qsub(RunSge):
+class qsub(RunJob):
 
     def __init__(self, config=None):
         '''
@@ -107,7 +107,7 @@ def main():
     if args.local:
         args.mode = "local"
     conf.update_dict(**args.__dict__)
-    logger = Mylog(logfile=args.log, level=args.debug and "debug" or "info")
+    logger = getlog(logfile=args.log, level=args.debug and "debug" or "info")
     qjobs = qsub(config=conf)
     try:
         qjobs.run(retry=args.resub, ivs=args.resubivs)
diff --git a/src/sge_run.py b/src/run.py
similarity index 99%
rename from src/sge_run.py
rename to src/run.py
index 0894e1e..18f6084 100644
--- a/src/sge_run.py
+++ b/src/run.py
@@ -18,7 +18,7 @@
 from .config import load_config, print_config
 
 
-class RunSge(object):
+class RunJob(object):
 
     def __init__(self, config=None):
         '''
@@ -629,9 +629,9 @@ def main():
         args.logdir = "runjob_"+os.path.basename(args.jobfile) + "_log_dir"
     args.logdir = os.path.join(args.workdir, args.logdir)
     conf.update_dict(**args.__dict__)
-    logger = Mylog(logfile=args.log,
-                   level="debug" if args.debug else "info", name=__name__)
-    runsge = RunSge(config=conf)
+    logger = getlog(logfile=args.log,
+                    level="debug" if args.debug else "info", name=__name__)
+    runsge = RunJob(config=conf)
     try:
         runsge.run(retry=args.resub, ivs=args.resubivs)
     except (JobFailedError, QsubError):
diff --git a/src/utils.py b/src/utils.py
index 482b993..07c1d59 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -169,7 +169,7 @@ def _run(self):
             raise e
 
 
-def Mylog(logfile=None, level="info", name=None):
+def getlog(logfile=None, level="info", name=None):
     logger = logging.getLogger(name)
     if level.lower() == "info":
         logger.setLevel(logging.INFO)