o
yf!3 @ s d dl Z d dlZd dlZd dlmZ d dlZd dlmZ d dl m
Z
d dl m Z G dd dZ
G dd d e
Zd
d Zh d ZG d
d dZeG dd dZG dd dZe ZdS ) N) dataclass)errors)cmd_opts) script_pathc @ sF e Zd ZdddZdd Zdd Zd
d Zd d
Zdd Zdd Z dS )
OptionInfoN Fc
C sV || _ || _|| _|| _|| _|| _| | _|| _d| _|| _ | | _
|
| _ | | _ d S )NF)
defaultlabel componentcomponent_argsonchangesection category_idrefresh do_not_savecomment_before
comment_afterinfotext restrict_api)
selfr r r
r r r
r r r r r r r `C:\Users\Luke\Documents\Stable diffusion\Automatic1111\stable-diffusion-webui\modules\options.py__init__ s zOptionInfo.__init__c C | j d| d| d7 _ | S )Nz
[]r )r r urlr r r link$ zOptionInfo.linkc C r )Nz
[r r )r r Zjs_funcr r r js( r z
OptionInfo.jsc C s | j d| d7 _ | S )Nz(z)r )r infor r r r! , s zOptionInfo.infoc C s | j |7 _ | S Nr )r htmlr r r r# 0 zOptionInfo.htmlc C | j d7 _ | S )Nz- (requires restart)r r r r r
needs_restart4 r$ zOptionInfo.needs_restartc C r% )Nz/ (requires Reload UI)r r& r r r needs_reload_ui8 r$ zOptionInfo.needs_reload_ui) Nr NNNNNr r NFN)
__name__
__module__ __qualname__r r r r! r# r' r( r r r r r
s
r c s e Zd Z fddZ ZS )
OptionHTMLc s( t jt| ddd d d| _d S )Nr c [ s t jdddi| S )N elem_classesz
settings-infor )grHTML)kwargsr r r ? z%OptionHTML.__init__..)r r
T)superr strstripr )r text __class__r r r > s
zOptionHTML.__init__)r) r* r+ r
__classcell__r r r7 r r, = s r, c C sJ | D ] }t| dkr| |_qt| dkr"| dd |_| d |_q|S )N r )valueslenr
r )Zsection_identifierZ options_dictvr r r options_sectionD s
r? > datatypemap data_labelsrestricted_optsc s e Zd ZeeiZdeeef fddZ fddZ
fddZ d d
d Z d d
Z
dd Zdd Zdd Zd ddZdd Zdd Zdd Zd d Z ZS ) OptionsrB c C s& || _ dd | j D | _|| _d S )Nc S s i | ] \}}|j s||jqS r )r r .0kr> r r r
W s z$Options.__init__..)rB itemsr@ rC )r rB rC r r r r U s
zOptions.__init__c sj |t v r
tt| ||S | jd ur|| jv s || jv rtjr#J d| j|d }|j r/d S |r4|j
nd }t |t rK|dddu rKt
d| dtjd urxtttjtjd}|jd }|jd }||vsxJ d| d
| d | d tjd urtttjtjd}||vsJ d| d
tjr|| jv rt
d| d|| j|< d S tt| ||S )Nz changing settings is disabledvisibleTFznot possible to set 'z' because it is restricted,r z ' because settings in section 'z' (z/) are frozen with --freeze-settings-in-sectionsz@' because this setting is frozen with --freeze-specific-settingsz4' because it is restricted with --hide_ui_dir_config)options_builtin_fieldsr3 rD __setattr__r@ rB r freeze_settingsgetr r
isinstancedict RuntimeErrorfreeze_settings_in_sectionslistmapr4 r5 splitr
freeze_specific_settingshide_ui_dir_configrC ) r keyvaluer! Z comp_argsZfrozen_sectionsZ section_key section_nameZ frozen_keysr7 r r rN Z s0
$
zOptions.__setattr__c s\ |t v r tt| |S | jd ur|| jv r| j| S || jv r&| j| jS tt| |S r" )rM r3 rD __getattribute__r@ rB r )r itemr7 r r __getattr__ s
zOptions.__getattr__FTc
C s | j |d}||kr
dS | j| }|jrdS |r |jr dS zt| || W n
ty0 Y dS w |re|jdurez| W dS tyd } zt
|d| d| t| || W Y d}~dS d}~ww dS )zhsets an option and calls its onchange callback, returning True if the option changed and False otherwiseNFzchanging setting z to T) r@ rP rB r r setattrrS r Exceptionr display)r rZ r[ Zis_apiZ
run_callbacksZoldvaloptioner r r set s0
z Options.setc C s | j |}|du r dS |jS )z%returns the default value for the keyN)rB rP r )r rZ
data_labelr r r get_default s zOptions.get_defaultc C sV t jrJ dt|ddd}tj| j|ddd W d d S 1 s$w Y d S )Nzsaving settings is disabledwutf8encoding F)indent ensure_ascii)r rO openjsondumpr@ )r filenamefiler r r save s "z Options.savec C sH |d u s|d u r
dS | j t|t|}| j t|t|}||kS )NT)rA rP type)r xyZtype_xZtype_yr r r same_type s
zOptions.same_typec C s z t |ddd}t|| _W d n1 sw Y W n. ty+ i | _Y n# tyM tjd| ddd t |tj
t dd i | _Y nw | j
d
d urh| j
d d u rh| j
d
| jd < | j
d d ur| j
d
d u rdd | j
d dD | jd
< t| j
dtr| j
drd| jvrdd | j
ddD | jd< d}| j D ]5\}}| j
|d }|d ur| |j|std| d| dt|j dt|jj d tjd |d7 }q|dkrtd | d tjd d S d S ) Nrri rj z*
Could not load settings
The config file "z^" is likely corrupted
It has been moved to the "tmp/config.json"
Reverting config to default
T)exc_infotmpz config.jsonZsd_vae_as_default&sd_vae_overrides_per_model_preferencesZ
quicksettingsZquicksettings_listc S g | ]}| qS r r5 rF ir r r
r2 z Options.load..rK Z
ui_reorderZui_reorder_listc S r} r r~ r r r r r r2 r z Warning: bad setting value: z: z (z ; expected ))rs rL zDThe program is likely to not work with bad settings.
Settings file: z/
Either fix the file, or delete it and restart.)ro rp loadr@ FileNotFoundErrorra r reportosreplacepathjoinr rP rW rQ r4 rI rB rx r printru r) sysstderr)r rr rs Z bad_settingsrG r> r! r r r r s8
"("6z Options.loadc C s$ | j |}||_|r| d S d S r" )rB rP r )r rZ funccallr^ r r r r s
zOptions.onchangec s fdd j D }dd j D |d< dd j D |d< i } j D ]%}|jd d u r5q+tj|j}|d u rBdn|j}||vrP|jd ||< q+d
d | D d d
gg |d< t
|S )Nc s" i | ]
\}}| j ||jqS r )r@ rP r rE r& r r rH " z$Options.dumpjson..c S " i | ]
\}}|j d ur||j qS r" r rE r r r rH r Z_comments_beforec S r r" r rE r r r rH r Z_comments_afterr Z
UncategorizedrL c S s g | ]\}}||gqS r r rE r r r r s z$Options.dumpjson..ZDefaultsOther _categories) rB rI r< r
categoriesmappingrP r r rp dumps)r dZitem_categoriesr^ categoryr r& r dumpjson s
zOptions.dumpjsonc C s2 || j |< || jvr|js|j| j|< d S d S d S r" )rB r@ r r )r rZ r! r r r
add_option s
zOptions.add_optionc s i i }| j }|D ]\}}|j|vr|j||j< q |D ] \}}||j|_q tjD ] }| vr8t |< q, fdd}tt ||d| _ dS )a# Reorder settings so that:
- all items related to section always go together
- all sections belonging to a category go together
- sections inside a category are ordered alphabetically
- categories are ordered by creation order
Category is a superset of sections: for category "postprocessing" there could be multiple sections: "face restoration", "upscaling".
This function also changes items' category_id so that all items belonging to a section have the same category_id.
c s, | d } |jt }|jd }||fS )NrL )rP r r= r
)rv r^ Zcategory_orderZ
section_orderZ category_idsr r sort_key s
z!Options.reorder..sort_key)rZ N)
rB rI r
r rP r r r= rR sorted)r Zsection_categoriesZsettings_items_r^ r r r r r reorder s
zOptions.reorderc C sd |du rdS | j | j}|du rt| |d}|du r dS t|}|tkr,|dkr,d}|S ||}|S )zcasts an arbitrary to the same type as this setting's value with key
Example: cast_value("eta_noise_seed_delta", "12") -> returns 12 (an int rather than str)
NFalseF)rB r getattrru bool)r rZ r[
default_value
expected_typer r r
cast_value) s zOptions.cast_value)FT)T)r) r* r+ intfloatrA rR r4 r r rN r_ re rg rt rx r r r r r r r9 r r r7 r rD R s )
$rD c @ s e Zd ZU eed< eed< dS )OptionsCategoryidr N)r) r* r+ r4 __annotations__r r r r r @ s
r c @ s e Zd Zdd Zdd ZdS )OptionsCategoriesc C s
i | _ d S r" )r r& r r r r F s
zOptionsCategories.__init__c C s" || j v r|S t||| j |< d S r" )r r )r r r r r r register_categoryI s
z#OptionsCategories.register_categoryN)r) r* r+ r r r r r r r E s r )r rp r dataclassesr gradior. modulesr modules.shared_cmd_optionsr modules.paths_internalr r r, r? rM rD r r r r r r r s$ 0 o