Compare commits

...

11 Commits

Author SHA1 Message Date
abae0a5fdb fix: 'common' version 2026-06-03 17:23:18 +08:00
a3e3c2a8fa fix: input cursor pointer 2026-05-27 17:48:54 +08:00
7cb59b5122 update: bump 'common' version to 0.1.15 2026-05-26 11:59:46 +08:00
b3fadc99f3 minor fix 2026-05-08 15:58:15 +08:00
c693d624fc minor fix 2026-05-08 15:28:12 +08:00
760b88faee minor fix 2026-05-08 15:10:16 +08:00
acb5607e2c minor fix 2026-05-08 14:26:17 +08:00
3007993ad2 minor fix 2026-05-08 10:35:47 +08:00
e578c31c28 minor fix 2026-05-08 10:18:12 +08:00
28a4252852 minor fix 2026-05-08 10:11:45 +08:00
eb6df01c06 fix: handling invoice pagination 2026-05-05 19:16:41 +08:00
4 changed files with 329 additions and 298 deletions

View File

@@ -128,166 +128,151 @@
</fieldset>
</form>
<script type="text/javascript">
const $ = (selectors, fn) => {
let e = document.querySelector(selectors);
return fn && e ? fn(e) : e;
};
<script type="module">
import { default as $, Rpc2, LogRecord } from '/';
if (performance.getEntriesByType('navigation')[0].type === 'reload') await Rpc2.invoke('exit');
const $$ = (selectors, fn) => {
let e = document.querySelectorAll(selectors);
return fn && e ? fn(e) : e;
};
let error = null;
let status = null;
let { profiles, parameters } = await Rpc2.invoke('context');
const $$$ = (selectors, event, listener) => {
$(selectors).addEventListener(event, listener);
};
let date = new Date();
let day = date.getDay() || 7;
$('#datefrom').valueAsNumber = date.setHours(-24 * (day - 1)) - date.getTimezoneOffset() * 60 * 1000;
$('#dateto').valueAsNumber = date.setHours(24 * 7) + date.getTimezoneOffset() * 60 * 1000;
$('#all').dispatchEvent(new Event('change'));
let Profiles = new Array();
let Latest = null;
let Status = null;
$.set('#begin', 'click', async () => {
switch (status) {
case 'READY':
let name = $('#name').value;
let options = { profile: name };
for (let element of $.all("input[type='checkbox']:not(.local-only)")) {
options[element.id] = element.checked;
}
function main(profiles, args) {
$$$('#begin', 'click', async () => {
switch (Status) {
case 'READY':
let name = $('#name').value;
let options = new Map();
for (let element of $$("input[type='checkbox']:not(.local-only)")) {
options[element.id] = element.checked;
}
for (let element of $.all("input[type='date']")) {
options[element.id] = element.value;
}
for (let element of $$("input[type='date']")) {
options[element.id] = element.value;
}
for (let element of $.all("input[type='number']")) {
parameters[element.id] = element.valueAsNumber;
}
for (let element of $$("input[type='number']")) {
args[element.id] = element.valueAsNumber;
}
options['profile'] = name;
await Rpc2.invoke('begin', options, args);
break;
case 'RUNNING':
await Rpc2.invoke('pause');
break;
case 'STANDBY':
await Rpc2.invoke('resume');
break;
}
});
$$$('#begin', 'click', (e) => {
$('#begin > span.icon').removeAttribute('hidden');
$('#begin > span.text').innerText = '';
$('#begin').classList.remove('pulse');
$('#begin').disabled = true;
});
$$$('#cancel', 'click', async (e) => {
$('#cancel').disabled = true;
await Rpc2.invoke('cancel');
});
$$$('#skip', 'click', async (e) => {
$('#skip').disabled = true;
await Rpcs.invoke('skip');
});
$$$('#logs', 'change', (e) => {
if (e.target.checked) $('#messages').removeAttribute('hidden');
else $('#messages').setAttribute('hidden', '');
});
$$$('#name', 'change', (e) => {
let p = Profiles.find(o => o.name === e.target.value);
for (let k of Object.keys(p)) $(`#${k}`, e => e.value = p[k] ?? '');
});
$$$('#all', 'change', (e) => {
$('#name').disabled = e.target.checked;
});
for (let item of JSON.parse(profiles)) {
Profiles.push(item);
$('#name').add(new Option(item.name, item.name));
$('#name').dispatchEvent(new Event('change'));
await Rpc2.invoke('begin', options, parameters);
break;
case 'RUNNING':
await Rpc2.invoke('pause');
break;
case 'STANDBY':
await Rpc2.invoke('resume');
break;
}
});
for (let item of $$("input[type='number']")) {
item.value = args[item.id];
}
$.set('#begin', 'click', () => {
$('#begin > span.icon').removeAttribute('hidden');
$('#begin > span.text').innerText = '';
$('#begin').classList.remove('pulse');
$('#begin').disabled = true;
});
let date = new Date();
let day = date.getDay() || 7;
$('#datefrom').valueAsNumber = date.setHours(-24 * (day - 1)) - date.getTimezoneOffset() * 60 * 1000;
$('#dateto').valueAsNumber = date.setHours(24 * 7) + date.getTimezoneOffset() * 60 * 1000;
$('#all').dispatchEvent(new Event('change'));
$.set('#cancel', 'click', async () => {
$('#cancel').disabled = true;
await Rpc2.invoke('cancel');
});
let account = new String(args['account']);
let name = account.split('@', 1).pop() ?? 'unknown';
$.set('#skip', 'click', async () => {
$('#skip').disabled = true;
await Rpcs.invoke('skip');
});
$.set('#logs', 'change', (e) => {
if (e.target.checked) $('#messages').removeAttribute('hidden');
else $('#messages').setAttribute('hidden', '');
});
$.set('#name', 'change', (e) => {
let p = profiles.find(o => o.name === e.target.value);
for (let k of Object.keys(p)) $(`#${k}`)?.setAttribute('value', p[k] ?? '');
});
$.set('#all', 'change', (e) => {
$('#name').disabled = e.target.checked;
});
if (parameters['account'] && parameters['password']) {
let account = new String(parameters['account']);
let name = account.split('@', 1).pop();
name = name.charAt(0).toLocaleUpperCase() + name.slice(1);
document.title += ` (${name})`;
}
setInterval(async () => {
let history = await Rpc2.invoke('history');
let logs = Array.from(history);
for (let item of profiles) {
$('#name').add(new Option(item.name, item.name));
$('#name').dispatchEvent(new Event('change'));
}
for (let record of logs) {
if (record.levelno >= 40) Latest = record;
let message = LogRecord.format(record);
let node = document.createTextNode(new String(message).concat('\n'));
$('#messages').appendChild(node);
$('#messages').scrollTop = $('#messages').scrollHeight;
}
for (let item of $.all("input[type='number']")) {
item.value = parameters[item.id];
}
Status = await Rpc2.invoke('status');
$('#statusLabel').innerText = Status.charAt(0).toUpperCase() + Status.slice(1).toLowerCase();
while (await new Promise(o => setTimeout(o, 1000, true))) {
let history = await Rpc2.invoke('history').catch(() => []);
let logs = Array.from(history);
switch (Status) {
case 'IDLE':
return;
case 'READY':
$('#begin > span.text').innerText = 'Begin';
$('#begin').classList.remove('pulse');
$('#progressLabel').innerText = '';
$('#remainingLabel').innerText = '';
break;
case 'RUNNING':
$('#begin > span.text').innerText = 'Pause';
$('#begin').classList.add('pulse');
for (let record of logs) {
if ($('#messages').childNodes.length >= 500) $('#messages').childNodes.item(0)?.remove();
if (record.levelno >= 40) error = record;
let message = LogRecord.format(record);
let node = document.createTextNode(new String(message).concat('\n'));
$('#messages').appendChild(node);
$('#messages').scrollTop = $('#messages').scrollHeight;
}
let progress = await Rpc2.invoke('progress');
let { task, number, index, limit } = progress;
$('#numberLabel').innerText = number ?? '';
$('#progressLabel').innerText = limit ? `${task}, ${parseFloat((index / limit * 100).toFixed(2))}% (${index}/${limit})` : task;
status = await Rpc2.invoke('status').catch(() => null);
$('#statusLabel').innerText = status ? status.charAt(0).toUpperCase() + status.slice(1).toLowerCase() : '';
let uptime = await Rpc2.invoke('uptime');
$('#uptimeLabel').innerText = Temporal.Duration.from({ seconds: uptime }).round({ largestUnit: 'hours' }).toLocaleString('en', { style: 'digital' });
switch (status) {
case 'IDLE':
continue;
case 'READY':
$('#begin > span.text').innerText = 'Begin';
$('#begin').classList.remove('pulse');
$('#progressLabel').innerText = '';
$('#remainingLabel').innerText = '';
break;
case 'RUNNING':
$('#begin > span.text').innerText = 'Pause';
$('#begin').classList.add('pulse');
if (index && limit && uptime) {
let rate = index / uptime;
let remaining = Math.floor((limit - index) / rate);
$('#remainingLabel').innerHTML = Temporal.Duration.from({ seconds: remaining }).round({ largestUnit: 'hours' }).toLocaleString('en');
}
break;
case 'STANDBY':
if (Latest !== null) {
alert(`(${Latest.levelname}) ${Latest.msg}\n${Latest.exc_text ?? ''}`);
Latest = null;
}
$('#begin > span.text').innerText = 'Resume';
$('#begin').classList.remove('pulse');
break;
}
$('#begin > span.icon').setAttribute('hidden', '');
$('#begin').disabled = false;
let progress = await Rpc2.invoke('progress').catch(() => new Object());
let { task, number, index, limit } = progress;
$('#numberLabel').innerText = number ?? '';
$('#progressLabel').innerText = limit ? `${task}, ${parseFloat((index / limit * 100).toFixed(2))}% (${index}/${limit})` : task;
let actions = await Rpc2.invoke('actions');
$('#cancel').disabled = !actions['Cancel'];
$('#skip').disabled = !actions['Skip'];
}, 1000);
let [t1, t2] = await Rpc2.invoke('uptime').catch(() => []);
$('#uptimeLabel').innerText = Temporal.Duration.from({ seconds: t1 ?? 0 }).round({ largestUnit: 'hours' }).toLocaleString('en', { style: 'digital' });
let remaining = index && limit && t2 ? Math.floor((limit - index) / (index / t2)) : 0;
$('#remainingLabel').innerHTML = remaining ? Temporal.Duration.from({ seconds: remaining }).round({ largestUnit: 'hours' }).toLocaleString('en') : '';
break;
case 'STANDBY':
if (error !== null) {
alert(`(${error.levelname}) ${error.msg}\n${error.exc_text ?? ''}`);
error = null;
}
$('#begin > span.text').innerText = 'Resume';
$('#begin').classList.remove('pulse');
break;
}
$('#begin > span.icon').setAttribute('hidden', '');
$('#begin').disabled = false;
let actions = await Rpc2.invoke('actions').catch(() => new Object());
$('#cancel').disabled = !actions['Cancel'];
$('#skip').disabled = !actions['Skip'];
}
</script>
@@ -299,6 +284,10 @@ body {
justify-content: center;
}
select, option, input[type='checkbox'], input[type='date'] {
cursor: pointer;
}
label {
user-select: none;
width: fit-content;
@@ -362,8 +351,8 @@ label {
}
@keyframes pulse {
50% {
opacity: 0.6;
}
50% {
opacity: 0.6;
}
}
</style>
</style>

348
main.py
View File

@@ -11,8 +11,9 @@ from selenium.common.exceptions import TimeoutException, NoSuchElementException
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from common import jsonrpc2
from common.utils import *
from common.timer import Timer
from common.jsonrpc2 import ServiceProvider
from common.actionflow import Action, ActionFlow
from enum import Enum
@@ -37,8 +38,9 @@ WEBURL = "https://crm.xiaoman.cn/%s"
APIURL = "https://%s.vosfactures.fr"
def main(driver: WebDriver, logger = logging.getLogger('main')):
http = PoolManager()
parameters = vars(args)
http = PoolManager()
sp = ServiceProvider.default()
class Status(Enum):
IDLE = 0
@@ -60,68 +62,67 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
result = ''.join(filter(bool, [self.prefix, number, self.suffix]))
return result
timer = Timer()
t1 = Timer()
t2 = Timer()
options = dict()
profiles: list[Profile] = list()
status = Status.IDLE
def begin(opts: dict, args: dict):
nonlocal options, status
nonlocal status
options.update(opts)
status = Status.RUNNING
parameters.update(args)
timer.clear()
timer.start()
t1.clear()
t1.start()
def pause():
nonlocal status
status = Status.STANDBY
timer.pause()
t1.pause()
t2.pause()
def resume():
nonlocal status
status = Status.RUNNING
driver.switch_to.window(driver.current_window_handle)
timer.start()
t1.start()
t2.start()
jsonrpc2.define('begin', begin)
jsonrpc2.define('pause', pause)
jsonrpc2.define('resume', resume)
jsonrpc2.define('status', lambda: status.name)
jsonrpc2.define('uptime', lambda: timer.delta())
sp.set('begin', begin)
sp.set('pause', pause)
sp.set('resume', resume)
sp.set('status', lambda: status.name)
sp.set('uptime', lambda: [t1.delta()])
try:
for source, profile in zip(parameters.get('profile'), repeat(dict())):
for key, value in map(lambda o: str.split(o, '=', 2), source):
profile[key.lower().strip()] = value.strip()
item = Profile(**profile)
profiles.append(item)
profiles = [
Profile(**{ k.lower().strip(): v.strip() for k, v in map(lambda o: str.split(o, '=', 2), p) })
for p in parameters['profile']
]
except Exception as e:
logger.critical('Unable to load profiles', exc_info=e)
return 2
try:
manifest = json.dumps(profiles, default=lambda o: vars(o))
driver.get(str(Path('index.html').resolve()))
driver.execute_script(jsonrpc2.prelude(), manifest, parameters)
sp.set('context', lambda: { 'profiles': list(map(vars, profiles)), 'parameters': parameters })
driver.get(sp.run())
except Exception as e:
logger.critical('Unable to load starup page', exc_info=e)
return 3
try:
driver.switch_to.new_window('tab')
driver.set_page_load_timeout(parameters.get('timeout'))
driver.set_page_load_timeout(parameters['timeout'])
driver.get(WEBURL % 'product')
except TimeoutException:
logger.warning('Timeout')
driver.execute_script("window.stop();")
setup(driver, parameters.get('attempts'), parameters.get('timeout'), parameters.get('interval'))
setup(driver, parameters)
until(lambda x: 'loginProgress' in x.find_element(By.TAG_NAME, "body").get_attribute('class'), watch=False)
logger.info('Waiting for authentication to complete...')
if (account := parameters.get('account')) and (password := parameters.get('password')):
if (account := parameters['account']) and (password := parameters['password']):
try:
logger.info('Logging in as %s (%s)', str.split(account, '@', 1).pop(0).capitalize(), account)
locate("input.account").send_keys(account)
@@ -138,7 +139,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.info('Done')
break
except:
sleep(parameters.get('interval'))
sleep(parameters['interval'])
class ProductInfo:
def __init__(self, file):
@@ -165,7 +166,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.info('Downloading product list...')
click("header .okki-space .okki-space-item:nth-child(1) button")
click(".okki-dropdown button")
sleep(parameters.get('interval'))
sleep(parameters['interval'])
click(".okki-modal.product-export-wrap .mm-selector-rendered")
click(".mm-outside.ui-field-selector-popper .selector-area:nth-child(1) button")
click(".okki-modal.product-export-wrap .okki-modal-footer button.okki-btn-primary")
@@ -173,14 +174,14 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
while True:
try:
click(".okki-modal.product-export-wrap .okki-modal-footer button.okki-btn-primary")
sleep(parameters.get('interval'))
sleep(parameters['interval'])
click(".okki-modal.product-export-wrap .okki-modal-body .virtual-list-wrap .vue-recycle-scroller__item-wrapper > div:nth-child(1) button.okki-btn-link", wait=False)
filename = locate(".okki-modal.product-export-wrap .okki-modal-body .virtual-list-wrap .vue-recycle-scroller__item-wrapper > div:nth-child(1) > div > div:nth-child(1) span").get_attribute('title')
break
except:
sleep(parameters.get('interval'))
sleep(parameters['interval'])
file = Path(parameters.get('directory')).joinpath(filename)
file = Path(parameters['directory']).joinpath(filename)
until(lambda _: file.exists(), watch=False)
p = ProductInfo(file)
driver.close()
@@ -191,30 +192,16 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.critical('Unable to load products', exc_info=e)
return 4
def wait(seconds: float):
while not sleep(seconds):
if status == Status.RUNNING:
break
def ready(driver: WebDriver):
html = driver.find_element(By.TAG_NAME, 'html')
return 'nprogress-busy' in html.get_attribute('class')
def fetch(url: str, method = 'GET', retry = parameters.get('attempts')):
def fetch(url: str, method = 'GET', retry = parameters['attempts']):
for attempt in range(1, retry + 1):
try:
response = http.request(method, url)
result = response.json()
return result
except Exception as e:
logger.error('Error while fetching data from %s, retrying... (%d)', url, attempt, exc_info=e)
logger.warning('Error while fetching data from %s, retrying... (%d)', url, attempt, exc_info=e)
assert attempt < retry
flow = ActionFlow()
profile = None
progress = { 'task': '' }
selection = 0
class Wait(Action):
@classmethod
def prepare(cls):
@@ -222,8 +209,18 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
@classmethod
def perform(cls):
flow.do(cls)
wait(1)
if status == Status.RUNNING: return False
sleep(0.2); return True
class Sleep(Action):
@classmethod
def prepare(cls):
return True
@classmethod
def perform(cls):
sleep(parameters['interval'])
return False
class Cancel(Action):
@classmethod
@@ -231,40 +228,56 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
nonlocal status
status = Status.RUNNING
return True
@classmethod
def perform(cls):
nonlocal status
status = Status.READY
driver.switch_to.window(driver.window_handles[0])
raise cls
class Skip(Action):
@classmethod
def prepare(cls):
return True
@classmethod
def perform(cls):
driver.switch_to.window(driver.current_window_handle)
raise cls
flow = ActionFlow()
flow.append(Wait)
flow.allow(Wait)
flow.do(Wait)
flow.append(Sleep)
flow.append(Cancel)
flow.append(Skip)
jsonrpc2.define('actions', lambda: flow.capabilities())
jsonrpc2.define('cancel', lambda: flow.do(Cancel))
jsonrpc2.define('skip', lambda: flow.do(Skip))
jsonrpc2.define('progress', lambda: progress)
profile = None
progress = { 'task': '' }
selection = 0
while not wait(1):
sp.set('actions', lambda: flow.capabilities())
sp.set('cancel', lambda: flow.do(Cancel, force=True))
sp.set('skip', lambda: flow.do(Skip, force=True))
sp.set('progress', lambda: progress)
while True:
try:
progress.clear()
progress['task'] = 'Task 1 of 4'
flow.allow(Cancel, False)
flow.allow(Skip, False)
flow.allow(Wait)
flow.allow(Sleep)
flow.do(Wait)
flow.react()
for i in range(len(driver.window_handles), 1, -1):
driver.switch_to.window(driver.window_handles[i-1])
driver.close()
driver.switch_to.window(driver.window_handles[i-2])
if options.get('all'):
profile = profiles[selection]
selection += 1
else:
name = options.pop('profile')
profile = next(filter(lambda o: o.name == name, profiles))
for i in range(len(driver.window_handles), 0):
driver.close()
driver.switch_to.window(driver.window_handles[i])
except (IndexError, KeyError):
logger.info('Done')
status = Status.READY
@@ -273,6 +286,17 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.error("Invalid profile '%s'", name)
status = Status.STANDBY
continue
except Exception as e:
logger.error('Unexpected error', exc_info=e)
status = Status.STANDBY
continue
progress.clear()
progress['task'] = 'Task 1 of 4'
t2.clear()
t2.start()
sp.pop('uptime')
sp.set('uptime', lambda: [t1.delta(), t2.delta()])
base = APIURL % profile.subdomain
data = list()
@@ -284,6 +308,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.info('Profile selected: %s', profile.name)
logger.info('Date from %s to %s', df, dt)
flow.allow(Cancel)
flow.allow(Skip, False)
for page in count(1):
try:
@@ -294,25 +319,27 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.info('Downloading invoices (%d)', page)
data.extend(result)
flow.do(Wait)
flow.do(Sleep)
flow.react()
wait(parameters.get('interval'))
except Skip:
pass
except Cancel:
status = Status.READY
break
except Exception as e:
logger.error('Error while fetching data from %s', base, exc_info=e)
break
if len(data) == 0:
logger.warning('Server returned empty response')
selection += 1
logger.warning('Server returned an empty response')
continue
logger.info('Initializing Workbook...')
progress['task'] = 'Task 2 of 4'
progress['limit'] = len(data)
progress['index'] = 0
t2.clear()
t2.start()
workbook = openpyxl.Workbook()
sheet = workbook.active
@@ -358,21 +385,26 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
try:
for i, item in enumerate(data, 1):
flow.do(Wait)
flow.react()
number: str = item['number']
logger.info('[%d/%d] Preprocessing data for %s', i, len(data), number)
progress['number'] = number
progress['index'] = i-1
if (id := item['category_id']) not in categories:
flow.react()
categories[id] = fetch(f'{base}/categories/{id}.json?api_token={profile.token}')
if (category := categories.get(o := item['category_id'])) is None:
if 'error' in (category := fetch(f'{base}/categories/{o}.json?api_token={profile.token}')):
error = category['error']
code = category['code']
logger.warning("Error while fetching 'category' (code: %s, message: %s); skipping", code, error)
continue
if (id := item['client_id']) not in clients:
flow.react()
clients[id] = fetch(f'{base}/clients/{id}.json?api_token={profile.token}')
category = categories.get(item['category_id'])
client = clients.get(item['client_id'])
if (client := clients.get(o := item['client_id'])) is None:
if 'error' in (client := fetch(f'{base}/clients/{o}.json?api_token={profile.token}')):
error = client['error']
code = client['code']
logger.warning("Error while fetching 'client' (code: %s, message: %s); skipping", code, error)
continue
identity = client['external_id'] if client['company'] else profile.person
date: str = item['issue_date']
@@ -417,11 +449,9 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
if record[9] is None: logger.warning("Could not identify product '%s'", product)
sheet.append(record.data)
flow.react()
except Skip:
pass
except Cancel:
status = Status.READY
continue
except Exception as e:
logger.error('Error while processing data', exc_info=e)
@@ -430,14 +460,14 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
try:
filename = f'Order-Import-{profile.name}-{datetime.now().strftime('%Y%m%d-%H%M%S-%f')}.xlsx'
file = Path(parameters.get('directory')).joinpath(filename)
file = Path(parameters['directory']).joinpath(filename)
logger.info('Saving excel file to %s', str(file))
flow.do(Wait)
flow.react()
workbook.save(file)
except Skip:
pass
except Cancel:
status = Status.READY
continue
except Exception as e:
logger.error('Error while saving excel file', exc_info=e)
@@ -448,6 +478,8 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.info('Uploading...')
progress.clear()
progress['task'] = 'Task 3 of 4'
t2.clear()
t2.start()
driver.switch_to.new_window('tab')
driver.get(WEBURL % 'order/importOrder')
@@ -455,6 +487,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
click(".product-import-img-box .mm-selector-rendered")
click(".mm-outside.mm-select-dropdown ul li:nth-child(%d) span" % (1 if options.get('draft') else 6))
flow.do(Wait)
flow.react()
locate(".big-file-upload input", wait=False).send_keys(str(file))
click(".product-import-img-footer button")
@@ -465,7 +498,8 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
innerText = driver.find_element(By.CSS_SELECTOR, ".img-box-result-title").get_attribute('innerText')
if innerText == '导入完成': break
except:
wait(parameters.get('interval'))
flow.do(Sleep)
flow.react()
click(".mm-notification-container .mm-icon-close")
click(".product-import-img-footer button.mm-button__primary")
@@ -474,16 +508,16 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
if err.get_attribute('disabled') is None:
logger.warning('Incomplete import detected; downloaded 1 related document')
err.click()
sleep(1)
flow.do(Sleep)
flow.react()
click(".mm-tbody table tbody tr:nth-child(1) td:nth-child(3) a")
driver.switch_to.window(driver.window_handles[2])
until(ready)
flow.do(Sleep)
flow.react()
logger.info('Done')
except Skip:
pass
except Cancel:
status = Status.READY
continue
except Exception as e:
logger.error('Error while uploading excel file', exc_info=e)
@@ -513,19 +547,29 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
progress['task'] = 'Task 4 of 4'
progress['limit'] = len(data)
t2.clear()
t2.start()
index = 0
attempts = 0
while index < len(data):
try:
attempts += 1
driver.switch_to.window(driver.window_handles[2])
item = data[index]
kind = item['kind']
title = re.search(r'O\d+', item['title'])
number = profile.format(item['number'])
positions = item['positions']
opportunity = None
attempts += 1
if attempts > parameters.get('attempts'):
if kind != 'vat':
logger.info('[%d/%d] Skipping %s', index+1, len(data), number)
index += 1
attempts = 0
continue
if attempts > parameters['attempts']:
logger.warning('Exhausted all allowed attempts; skipping %s', number)
index += 1
attempts = 0
@@ -535,6 +579,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
progress['index'] = index
flow.allow(Cancel)
flow.allow(Skip)
flow.do(Wait)
flow.react()
try:
@@ -553,17 +598,20 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
'array_flag': 0
}])
driver.get(url.encode())
wait(parameters.get('interval'))
flow.do(Wait)
flow.do(Sleep)
flow.react()
url = Parse(driver.current_url)
if url.get('page') != page: raise Exception(number)
wait(parameters.get('interval'))
flow.do(Sleep)
flow.react()
link = locate(".virtual-list-wrap .vue-recycle-scroller .vue-recycle-scroller__item-wrapper > div:nth-child(1) .cell[data-cci='1'] a", wait=False)
if link.text != number: continue
logger.info('[%d/%d] Processing %s...', index+1, len(data), number)
link.click()
driver.switch_to.window(driver.window_handles[3])
flow.do(Wait)
flow.react()
click(".sticky .okki-space-item:nth-child(1) button")
break
@@ -572,7 +620,6 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
attempts = 0
continue
except Cancel:
status = Status.READY
break
except NoSuchElementException:
logger.warning("Could not find invoice '%s'; skipping", number)
@@ -589,7 +636,8 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
driver.switch_to.new_window('tab')
url = Parse(WEBURL % 'crm/business/list')
driver.get(url.encode({ 'mode': 'list' }))
wait(parameters.get('interval'))
flow.do(Wait)
flow.do(Sleep)
flow.react()
try: click(".new-wrapper .paas-next-invoice-list-filter-line-wrapper .okki-btn-background-ghost", wait=False)
@@ -602,12 +650,14 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
url.set('curPage', page)
url.set('pageSize', 1)
driver.get(url.encode({ 'keyword': match, 'search_field': 'serial_keyword' }))
wait(parameters.get('interval'))
flow.do(Wait)
flow.do(Sleep)
flow.react()
url = Parse(driver.current_url)
if url.get('curPage') != page: raise Exception(match)
wait(parameters.get('interval'))
flow.do(Sleep)
flow.react()
cell = locate(".virtual-list-wrap .vue-recycle-scroller .row-item > .cell:nth-child(3) .ow-serial-read-pretty_ellipsis", wait=False)
if cell.text != match: continue
link = locate(".virtual-list-wrap .vue-recycle-scroller .row-item > .cell:nth-child(6) a", wait=False)
@@ -618,7 +668,6 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
attempts = 0
continue
except Cancel:
status = Status.READY
break
except NoSuchElementException:
logger.warning("Could not find opportunity '%s'; skipping", match)
@@ -632,6 +681,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
if opportunity is not None:
try:
flow.do(Wait)
flow.react()
dropdown = locate("#rc_select_1")
dropdown.clear()
@@ -648,86 +698,95 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
attempts = 0
continue
except Cancel:
status = Status.READY
break
except Exception as e:
logger.warning('Error while selecting opportunity; skipping', exc_info=e)
logger.warning('Could not select opportunity; skipping', exc_info=e)
try:
pagination = 10
click(".okki-pagination-options-size-changer")
click(".okki-select-dropdown .rc-virtual-list-holder-inner > div:nth-child(1)")
except:
logger.warning('Unable to setup pagination; this may cause issues')
pids = list()
wrapper = locate(".paas-order-product-list .row-items", condition=None)
class Eureka(Exception): pass
try:
ids = list()
wrapper = locate(".paas-order-product-list .row-items", condition=None)
for page in count(1):
hits = 0
iteration = 0
flow.do(Wait)
flow.react()
tail = next(reversed(pids), None)
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", wrapper)
driver.execute_script("arguments[0].scrollTo(arguments[1], arguments[2]);", wrapper, 0, 0)
for item in wrapper.find_elements(By.CSS_SELECTOR, ".row-item"):
if len(pids) >= len(positions): raise Eureka()
div = item.find_element(By.CSS_SELECTOR, ".cell[data-cci='2'] .cell-inner div")
while hits < pagination and iteration < parameters['attempts']:
iteration += 1
height = int(wrapper.get_attribute('clientHeight')) if iteration > 1 else 0
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", wrapper)
driver.execute_script("arguments[0].scrollTo(0, arguments[1]);", wrapper, height)
rows = wrapper.find_elements(By.CSS_SELECTOR, ".row-item")
if (serial := int(div.text)) and serial not in pids:
for row in reversed(rows) if iteration > 1 else rows:
flow.do(Wait)
flow.react()
base = (serial - tail) if tail is not None else index
height = int(item.get_attribute('clientHeight'))
offset = (base - 1) * height
driver.execute_script("arguments[0].scrollTo(arguments[1], arguments[2]);", wrapper, 0, offset)
#span = item.find_element(By.CSS_SELECTOR, ".cell[data-cci='3'] .product-info-group-product-info:nth-child(3) span")
#code = span.text[6:]
value = item.find_element(By.CSS_SELECTOR, ".cell[data-cci='4'] input").get_attribute('value')
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", wrapper)
driver.execute_script("arguments[0].scroll(arguments[1], arguments[2]);", wrapper, 400, offset)
wait(parameters.get('interval'))
target = item.find_element(By.CSS_SELECTOR, ".cell[data-cci='6'] input")
driver.execute_script("arguments[0].scrollTo(0, arguments[1]);", wrapper, height)
serial = row.text.split('\n', 1)[0].strip()
if not serial or serial in ids: continue
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", row)
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", wrapper)
value = row.find_element(By.CSS_SELECTOR, ".cell[data-cci='4'] input").get_attribute('value')
driver.execute_script("arguments[0].scroll(400, 0);", wrapper)
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", row)
driver.execute_script("arguments[0].scrollIntoView({ block: 'center' });", wrapper)
flow.do(Sleep)
flow.react()
target = row.find_element(By.CSS_SELECTOR, ".cell[data-cci='6'] input")
if (target.get_attribute('value') == '0'):
target.send_keys(Keys.BACKSPACE)
target.send_keys(value)
pids.append(serial)
ids.append(serial)
hits += 1
button = locate(".text-right li.okki-pagination-next button", condition=None)
if button.get_attribute('disabled') is not None and len(pids) < len(positions):
raise Exception('Product list imcomplete; expected %d, got %d', len(pids), len(positions))
flow.react()
click(button)
except Eureka:
pass
if len(ids) >= len(positions): break
button = locate(".text-right li.okki-pagination-next button", condition=None)
if button.get_attribute('disabled') is not None and len(ids) < len(positions):
raise Exception('Product list imcomplete; expected %d, got %d' % (len(positions), len(ids)))
flow.do(Wait)
flow.react()
click(button)
except Skip:
index += 1
attempts = 0
continue
except Cancel:
status = Status.READY
break
except Exception as e:
logger.error('Error while modifying invoice', exc_info=e)
status = Status.STANDBY
continue
try: click(".ow-box button.okki-btn-round", wait=False)
except: pass
try:
click(".ow-box button.okki-btn-round", wait=False)
flow.do(Sleep)
flow.react()
except Exception as e:
logger.warning('Unable to unset additional fees; skipping', exc_info=e)
try:
wait(parameters.get('interval'))
flow.do(Wait)
flow.react()
flow.allow(Cancel, False)
flow.allow(Skip, False)
click(".sticky.bottom-0 button.okki-btn-primary", condition=None)
wait(parameters.get('interval'))
flow.do(Sleep)
flow.react()
except Skip:
pass
except Cancel:
status = Status.READY
break
except Exception as e:
logger.error('Error while saving document', exc_info=e)
@@ -735,7 +794,6 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
continue
finally:
driver.close()
driver.switch_to.window(driver.window_handles[2])
index += 1
attempts = 0
@@ -743,16 +801,6 @@ def main(driver: WebDriver, logger = logging.getLogger('main')):
logger.error('Unexpected error', exc_info=e)
status = Status.STANDBY
try:
driver.close()
driver.switch_to.window(driver.window_handles[1])
driver.close()
driver.switch_to.window(driver.window_handles[0])
selection += 1
except Exception as e:
logger.error('Unexpected error', exc_info=e)
status = Status.STANDBY
if __name__ == '__main__':
try:
logging.basicConfig(level=logging.INFO, format="[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s", datefmt="%Y-%m-%d %H:%M")
@@ -760,14 +808,6 @@ if __name__ == '__main__':
level = logging.getLevelNamesMapping().get(args.log_level, 'INFO')
logger.setLevel(level)
logger.info('Initializing...')
history = jsonrpc2.History()
logger.addHandler(history)
opts = jsonrpc2.Options()
jsonrpc2.define('history', lambda: history.truncate())
jsonrpc2.run(opts)
logger.info('Creating automation instance')
opts = ChromeOptions()
opts.enable_downloads = True
opts.add_argument('--deny-permission-prompts')
@@ -776,6 +816,8 @@ if __name__ == '__main__':
with keep.presenting():
driver = Chrome(options=opts)
status = main(driver)
except KeyboardInterrupt:
status = 0
except Exception as e:
logger.critical('Fatal error', exc_info=e)
status = 1

View File

@@ -1,4 +1,4 @@
cd /D .\..\
set SE_PROXY=http://127.0.0.1:10809
.\venv\Scripts\pythonw.exe .\main.py "user@example.com" "example" --profile "Example-1" --token "TOKEN" --subdomain "eg" --remise "100"
.\venv\Scripts\pythonw.exe .\main.py "user@example.com" "example" --profile "name=example" "token=TOKEN" "subdomain=eg"
@pause

Binary file not shown.